Zero sret copy

EnzymeAD · Dec 16, 2023 · 65fd8ac · 65fd8ac
1 parent d22dd49
commit 65fd8ac
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/enzyme/Enzyme/CApi.cpp b/enzyme/Enzyme/CApi.cpp
@@ -1683,6 +1683,17 @@ void EnzymeFixupJuliaCallingConvention(LLVMValueRef F_C) {
                              ArrayRef<int> inds, int sretCount, Type *ptrTy) {
           if (auto PT = dyn_cast<PointerType>(curType)) {
             if (PT->getAddressSpace() == 10) {
+              SmallVector<Value *, 1> outinds;
+              auto c0 = ConstantInt::get(B.getInt64Ty(), 0);
+              outinds.push_back(c0);
+              if (sretCount >= 0)
+                outinds.push_back(ConstantInt::get(B.getInt32Ty(), sretCount));
+              for (auto v : inds) {
+                outinds.push_back(ConstantInt::get(B.getInt32Ty(), v));
+              }
+              if (outinds.size() > 1)
+                out = B.CreateInBoundsGEP(sretTy, out, outinds);
+              B.CreateStore(ConstantPointerNull::get(PT), out);
               return;
             }
           }