Bump IREE and fix cpu_comparison test (nod-ai#429)

erwei-xilinx · Jun 17, 2024 · 07070f7 · 07070f7
1 parent 5d91be6
commit 07070f7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 5 deletions.
diff --git a/build_tools/ci/cpu_comparison/run_test.sh b/build_tools/ci/cpu_comparison/run_test.sh
@@ -262,9 +262,6 @@ function run_test() {
   fi
   echo "**** Running test for ${test_file} ****"
 
-  # Disable exit on failure:
-  set +e
-
   aie_vmfb="${OUTPUT_DIR}/${name}_aie.vmfb"
   cpu_vmfb="${OUTPUT_DIR}/${name}_cpu.vmfb"
 
@@ -283,6 +280,7 @@ function run_test() {
   echo "**** Generating CPU .vmfb file ****"
   ${IREE_COMPILE_EXE} "${test_file}"  \
       --iree-hal-target-backends=llvm-cpu \
+      --iree-llvmcpu-target-cpu-features=host \
       -o "${cpu_vmfb}"
 
   # Load the contents of OUTPUT_DIR/{name}_command_args.txt into a variable:
@@ -307,7 +305,7 @@ run_test \
    --name_prefix "matmul" \
    --lhs_rhs_type "bf16" \
    --acc_type "f32" \
-   --m "8"  --n "32" --k "64" \
+   --m "32"  --n "32" --k "64" \
    --rtol 1e-10 \
    --atol 1e-10
 

diff --git a/sync_deps.py b/sync_deps.py
@@ -7,7 +7,7 @@
 ### Update with: shark-workspace pin
 
 PINNED_VERSIONS = {
-  "iree": "e52b8bf3d4c1e09cca661bc5345a4c0dce5a6f54",
+  "iree": "42ac74291a81f495d9b34fcf444bf4b23b5a7966",
 }
 
 ORIGINS = {