Update IREE deps (iree-org#880)

IREE deps were previously pinned to `3.5.0rc20250516` as
`3.5.0rc20250517` introduced regressions. There were 2 issues:
* Shared memory handling changes in IREE were causing the crash as we
are were trying to allocate too much shared mem. IREE doesn't use
dynamic shared mem anymore so we need to pass 0 now in wave runtime
(test iree-org#855).
* 40% perf regression in extend attention caused by some changes in llvm
AMDGPU backend. These seems to went away after the recent IREE llvm
intergrate. I didn't dig too deep on them but I suspect
llvm/llvm-project#137807 was the issue (fixed by
llvm/llvm-project#140921).

---------

Signed-off-by: Ivan Butygin <[email protected]>
Signed-off-by: xintin <[email protected]>

Author: Hardcode84

iree/turbine/kernel/wave/runtime/runtime.cpp

@@ -88,7 +88,7 @@ static int launch(const KernelLaunchInfo &info, const Int64Vector &tensors,
         HIP_LAUNCH_PARAM_END};
 
     HIP_CHECK_RETURN(hipModuleLaunchKernel(function, info.gridX, info.gridY, info.gridZ,
-                                           info.blockX, info.blockY, info.blockZ, info.sharedMemoryBytes,
+                                           info.blockX, info.blockY, info.blockZ, 0,
                                            stream, nullptr, (void **)&hipLaunchParams));
 
     return hipSuccess;

lit_tests/kernel/wave/casting.py

@@ -76,7 +76,7 @@ def test(
         a_reg = tkw.cast(a_reg, tkl.f16)
         tkw.write(a_reg, b, elements_per_thread=16)
 
-    options = get_wave_compile_options(canonicalize=True)
+    options = get_wave_compile_options(canonicalize=False)
     test = wave_compile(options, test)
     print(test.asm)
 

requirements-iree-pinned.txt

@@ -7,7 +7,5 @@
 # Uncomment to skip versions from PyPI (so _only_ nightly versions).
 # --no-index
 
-# TODO: We have a pretty bad regression in Wave between 3.5.0rc20250516 and
-# 3.5.0rc20250517. I'm locking to 3.5.0rc20250516 until further investigation.
-iree-base-compiler==3.5.0rc20250516
-iree-base-runtime==3.5.0rc20250516
+iree-base-compiler==3.5.0rc20250530
+iree-base-runtime==3.5.0rc20250530