Fix intermediate blascall (#1954)

EnzymeAD · Jun 30, 2024 · 8cc86db · 8cc86db
1 parent 9e02236
commit 8cc86db
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp b/enzyme/tools/enzyme-tblgen/blas-tblgen.cpp
@@ -1105,9 +1105,12 @@ void rev_call_arg(bool forward, DagInit *ruleDag, const TGPattern &pattern,
          << "    }\n\n";
       os << "    auto cubcall = cast<CallInst>(Builder2.CreateCall(derivcall_"
          << dfnc_name << ", marg, Defs));\n";
-      os << "         SmallVector<Value*, 1>(1, cublasv2 ? "
+      os << "         SmallVector<Value*, 1> resvec(1, cublasv2 ? "
          << " (Value*)Builder2.CreateLoad(fpType, marg[marg.size()-1]) : "
-            "(Value*)cubcall);\n";
+            "(Value*)cubcall);\n"
+         << "         resvec[0] = to_blas_fp_callconv(Builder2, resvec[0], "
+            "byRefFloat, blasFPType, allocationBuilder, \"blascall\");\n"
+         << "         resvec;\n";
       os << " })\n";
       return;
     }