Don't store undef value sret

EnzymeAD · Oct 22, 2024 · 9da1d3e · 9da1d3e
1 parent 1a22934
commit 9da1d3e
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/enzyme/Enzyme/CApi.cpp b/enzyme/Enzyme/CApi.cpp
@@ -1966,6 +1966,12 @@ void EnzymeFixupJuliaCallingConvention(LLVMValueRef F_C) {
             outinds.push_back(ConstantInt::get(B.getInt32Ty(), v));
           }
 
+          if (isa<UndefValue>(in) || isa<PoisonValue>(in))
+            return;
+
+          if (isa<UndefValue>(out) || isa<PoisonValue>(out))
+            return;
+
           if (outinds.size() > 1)
             out = B.CreateInBoundsGEP(sretTy, out, outinds);
           if (ininds.size() > 1)