Add F16 support for benchmark. (iree-org#18580)

Add F16 support for the benchmark. Signed-off-by: erman-gurses <erman@nod-labs.com>
nod-ai · Sep 25, 2024 · 7290283 · 7290283
1 parent 3773a48
commit 7290283
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/runtime/bindings/python/iree/runtime/benchmark.py b/runtime/bindings/python/iree/runtime/benchmark.py
@@ -37,6 +37,7 @@
     numpy.dtype(numpy.int64): "i64",
     numpy.dtype(numpy.float64): "f64",
     numpy.dtype(numpy.int16): "i16",
+    numpy.dtype(numpy.float16): "f16",
     numpy.dtype(numpy.int8): "i8",
     numpy.dtype(numpy.bool_): "i1",
 }