Weird fix

nod-ai · Feb 13, 2025 · 6b3124d · 6b3124d
1 parent 3ce36bb
commit 6b3124d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/shortfin/python/array_binding.cc b/shortfin/python/array_binding.cc
@@ -202,6 +202,7 @@ class Refs {
     add_type(DType::int64(), "q", sizeof(signed long long));
     add_type(DType::sint64(), "q", sizeof(signed long long));
     add_type(DType::uint64(), "Q", sizeof(unsigned long long));
+    add_type(DType::bfloat16(), "H", sizeof(unsigned short));
     add_type(DType::float16(), "H", sizeof(unsigned short));
     add_type(DType::float32(), "f", sizeof(float));
     add_type(DType::float64(), "d", sizeof(double));