[SYCL] Fix invalid interator after removal in SYCLJointMatrixTransform (

#16134) Can be exposed in windows debug build on the tests added by the original patch: LLVM::SYCLLowerIR/JointMatrixTransform/access-chain-no-uses.ll Author: Jinsong Ji <[email protected]> Signed-off-by: Sidorov, Dmitry <[email protected]>
intel · Nov 20, 2024 · 8ad42d5 · 8ad42d5
1 parent d9c7bcb
commit 8ad42d5
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/llvm/lib/SYCLLowerIR/SYCLJointMatrixTransform.cpp b/llvm/lib/SYCLLowerIR/SYCLJointMatrixTransform.cpp
@@ -26,8 +26,9 @@ static constexpr char MATRIX_TYPE[] = "spirv.CooperativeMatrixKHR";
 // its users and operands to make LLVM IR more SPIR-V friendly.
 bool transformAccessChain(Function *F) {
   bool ModuleChanged = false;
-  for (auto I : F->users()) {
-    auto *CI = dyn_cast<CallInst>(I);
+  for (auto I = F->user_begin(), E = F->user_end(); I != E;) {
+    User *U = *I++;
+    auto *CI = dyn_cast<CallInst>(U);
     if (!CI)
       continue;