Add MIOpen conv2D bf16 kernels for benchmarking (#34)

nod-ai · Nov 19, 2024 · 36f8d1e · 36f8d1e
1 parent 32fb07d
commit 36f8d1e
Show file tree

Hide file tree

Showing 3 changed files with 940 additions and 2 deletions.
diff --git a/convbench/conv_utils.py b/convbench/conv_utils.py
@@ -126,7 +126,7 @@ def generate_mlir(config: ConvConfig):
         out = str(n) + "x" + str(f) + "x" + str(h) + "x" + str(w) + "x" + str(elem_types[2])
     one = "1"
     zero = "0"
-    if (elem_types[0][0] == "f"):
+    if (elem_types[0][0] == "f" or elem_types[0][0] == "b"):
         one = "1.0"
         zero = "0.0"
     conv_template = CONV