Fix MLP benchmark config (#949)

Adds missing mlir-gen flags.
plaidml · Aug 6, 2024 · de30485 · de30485
1 parent 92cb3e1
commit de30485
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/benchmarks/config/base/base.json b/benchmarks/config/base/base.json
@@ -107,7 +107,7 @@
   "mlp_models": {
     "fp32_3x1024_const_mlir": {
       "type": "IR-GEN",
-      "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ],
+      "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ],
       "environment": {},
       "flags": [ "-n", "100" ],
       "extensions": [ "(avx2|asimd)" ]

diff --git a/benchmarks/config/base/named-ops.json b/benchmarks/config/base/named-ops.json
@@ -3,7 +3,7 @@
   "mlp_named_ops": {
     "fp32_3x1024_const_mlir": {
       "type": "IR-GEN",
-      "benchmark": [ "mlir-gen", "--output=named --kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ],
+      "benchmark": [ "mlir-gen", "--output=named --kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ],
       "environment": {},
       "flags": [ "-n", "100" ],
       "extensions": [ "(avx2|asimd)" ]