fix

Signed-off-by: Sidorov, Dmitry <[email protected]>

Author: MrSidims

clang/lib/Driver/ToolChains/SYCL.cpp

@@ -1720,7 +1720,7 @@ void SYCLToolChain::AddImpliedTargetArgs(const llvm::Triple &Triple,
     // -foffload-fp32-prec-div JIT
     Args.AddLastArg(BeArgs, options::OPT_foffload_fp32_prec_div);
     // -foffload-fp32-prec-sqrt JIT
-    Args.AddLastArg(BeArgs, options::OPT_OPT_foffload_fp32_prec_sqrt);
+    Args.AddLastArg(BeArgs, options::OPT_foffload_fp32_prec_sqrt);
   }
   if (IsGen) {
     for (auto [DeviceName, BackendArgStr] : PerDeviceArgs) {

llvm/lib/SYCLLowerIR/SYCLSqrtFDivMaxErrorCleanUp.cpp

@@ -106,6 +106,7 @@ SYCLSqrtFDivMaxErrorCleanUpPass::run(Module &M,
   // Replace @llvm.fpbuiltin.sqrt call with @llvm.sqrt. llvm-spirv will handle
   // it later.
   SmallSet<Function *, 2> DeclToRemove;
+  SmallVector<IntrinsicInst *, 16> InstsToRemove;
   for (auto *Sqrt : WorkListSqrt) {
     DeclToRemove.insert(Sqrt->getCalledFunction());
     IRBuilder Builder(Sqrt);
@@ -125,8 +126,7 @@ SYCLSqrtFDivMaxErrorCleanUpPass::run(Module &M,
     FastMathFlags FMF = FPOp->getFastMathFlags();
     NewSqrt->setFastMathFlags(FMF);
     Sqrt->replaceAllUsesWith(NewSqrt);
-    Sqrt->dropAllReferences();
-    Sqrt->eraseFromParent();
+    InstsToRemove.push_back(Sqrt);
   }
 
   // Replace @llvm.fpbuiltin.fdiv call with fdiv.
@@ -144,8 +144,13 @@ SYCLSqrtFDivMaxErrorCleanUpPass::run(Module &M,
     FastMathFlags FMF = FPOp->getFastMathFlags();
     NewFDiv->setFastMathFlags(FMF);
     FDiv->replaceAllUsesWith(NewFDiv);
-    FDiv->dropAllReferences();
-    FDiv->eraseFromParent();
+    InstsToRemove.push_back(FDiv);
+  }
+
+  // Clear instructions.
+  for (auto *Inst : llvm::reverse(InstsToRemove)) {
+    Inst->dropAllReferences();
+    Inst->eraseFromParent();
   }
 
   // Clear old declarations.