Bump mlir-air to 3d1a4e19ff748897a37e0eb88b59999197dbc0f8 (nod-ai#868)

newling · Nov 1, 2024 · 9aeeeb8 · 9aeeeb8
1 parent c6156e3
commit 9aeeeb8
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 12 deletions.
diff --git a/build_tools/ci/run_matmul_test.sh b/build_tools/ci/run_matmul_test.sh
@@ -555,17 +555,16 @@ run_matmul_test \
 # MLIR-AIR Matmul tests
 ###################################################################
 
-# TODO: re-enable after fixing in AIR
-# if [ -d "$VITIS" ]; then
-#   run_matmul_test \
-#       --name_prefix "ukern" \
-#       --lower_to_aie_pipeline "air" \
-#       --tile_pipeline "pad-pack" \
-#       --lhs_rhs_type "bf16" \
-#       --acc_type "f32" \
-#       --m "256"  --k "256" --n "256" \
-#       --use_ukernel "1"
-# fi
+if [ -d "$VITIS" ]; then
+  run_matmul_test \
+      --name_prefix "ukern" \
+      --lower_to_aie_pipeline "air" \
+      --tile_pipeline "pad-pack" \
+      --lhs_rhs_type "bf16" \
+      --acc_type "f32" \
+      --m "256"  --k "256" --n "256" \
+      --use_ukernel "1"
+fi
 
 # Example of a run with a group of 2+ matmuls. Currently this test is passed
 # the flag '--num_repeat_runs 0" as there is currently an issue with the runtime if
@@ -593,20 +592,23 @@ run_matmul_test \
 
 run_matmul_test \
     --name_prefix "packPeel_i32" \
+    --lower_to_aie_pipeline "air" \
     --tile_pipeline "pack-peel" \
     --lhs_rhs_type "i32" \
     --acc_type "i32" \
     --m "64"  --n "64" --k "128"
 
 run_matmul_test \
     --name_prefix "packPeel_bf16" \
+    --lower_to_aie_pipeline "air" \
     --tile_pipeline "pack-peel" \
     --lhs_rhs_type "bf16" \
     --acc_type "f32" \
     --m "512"  --n "512" --k "512"
 
 run_matmul_test \
   --name_prefix "packPeel_t_bf16" \
+  --lower_to_aie_pipeline "air" \
   --tile_pipeline "pack-peel" \
   --lhs_rhs_type "bf16" \
   --acc_type "f32" \

diff --git a/third_party/mlir-air b/third_party/mlir-air