Fix refactor

nod-ai · Feb 17, 2025 · daf06a6 · daf06a6
1 parent 3ba3f60
commit daf06a6
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/shortfin/python/shortfin_apps/utils.py b/shortfin/python/shortfin_apps/utils.py
@@ -9,7 +9,9 @@
     "bfloat16": "bf16",
     "float32": "f32",
     "float16": "f16",
+    sfnp.int8: "i8",
     sfnp.float32: "f32",
+    sfnp.float16: "fp16",
     sfnp.bfloat16: "bf16",
 }