Correctly handle uncanonicalized addrspacecasts.

JuliaLang · Mar 15, 2024 · cdaeac9 · cdaeac9
1 parent e3c28a5
commit cdaeac9
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/llvm-remove-addrspaces.cpp b/src/llvm-remove-addrspaces.cpp
@@ -208,7 +208,14 @@ bool RemoveNoopAddrSpaceCasts(Function *F)
  LLVM_DEBUG(
  dbgs() << "Removing noop address space cast:\n"
  << I << "\n");
- ASC->replaceAllUsesWith(ASC->getOperand(0));
+ if (ASC->getType() == ASC->getOperand(0)->getType()) {
+ ASC->replaceAllUsesWith(ASC->getOperand(0));
+ } else {
+ // uncanonicalized addrspacecast; demote to bitcast
+ llvm::IRBuilder<> builder(ASC);
+ auto BC = builder.CreateBitCast(ASC->getOperand(0), ASC->getType());
+ ASC->replaceAllUsesWith(BC);
+ }
  NoopCasts.push_back(ASC);
  }
  }