Add env flag

slyalin · Aug 9, 2024 · 61fbde9 · 61fbde9
1 parent c20bdd0
commit 61fbde9
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 1 deletion.
diff --git a/tools/mlir_bench/ov_raw_mlir_bench.sh b/tools/mlir_bench/ov_raw_mlir_bench.sh
@@ -99,7 +99,7 @@ for MB in "${MINI_BATCHES[@]}"; do
     echo "MODEL_CONFIG=${MODEL_CONFIG}"
     GEN_FLAGS=(-t ${DATA_TYPE} -n ${MODEL_NAME})
     GEN_FLAGS+=(-p)
-    ENV_FLAGS=OV_MLIR_TPP=0
+    ENV_FLAGS="OV_MLIR_TPP=0 OV_MLIR_DEBUG=1"
     MODEL_OUT=$(exec env ${ENV_FLAGS} python3 ${MODEL_GEN} "${MODEL_CONFIG[@]}" "${GEN_FLAGS[@]}" 2>&1)
     if [ $? != 0 ]; then
         echo "Failed to generate model"

diff --git a/tools/mlir_bench/run_bench_bf16.sh b/tools/mlir_bench/run_bench_bf16.sh
@@ -24,6 +24,8 @@ if [ ! "${NUM_LAYERS}" ]; then
   NUM_LAYERS=3
 fi
 
+export OV_MLIR_DEBUG=1
+
 echo "### MLP BF16 benchmarks ###"
 echo "# Layers: ${NUM_LAYERS} #"
 echo "LIBXSMM"

diff --git a/tools/mlir_bench/run_bench_f32.sh b/tools/mlir_bench/run_bench_f32.sh
@@ -24,6 +24,8 @@ if [ ! "${NUM_LAYERS}" ]; then
   NUM_LAYERS=3
 fi
 
+export OV_MLIR_DEBUG=1
+
 echo "### MLP F32 benchmarks ###"
 echo "# Layers: ${NUM_LAYERS} #"
 echo "LIBXSMM"