Add will-return attribute for GPU compilation

exaloop · Nov 10, 2023 · 063ef3a · 063ef3a
1 parent effd412
commit 063ef3a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/codon/cir/llvm/gpu.cpp b/codon/cir/llvm/gpu.cpp
@@ -501,8 +501,10 @@ void moduleToPTX(llvm::Module *M, const std::string &filename,
   remapFunctions(M);
 
   // Strip debug info and remove noinline from functions (added in debug mode).
+  // Also, tell LLVM that all functions will return.
   for (auto &F : *M) {
     F.removeFnAttr(llvm::Attribute::AttrKind::NoInline);
+    F.setWillReturn();
   }
   llvm::StripDebugInfo(*M);