fix mem issue

EnzymeAD · Oct 16, 2023 · e358c7d · e358c7d
1 parent a406e3c
commit e358c7d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/enzyme/Enzyme/Utils.cpp b/enzyme/Enzyme/Utils.cpp
@@ -680,8 +680,8 @@ void callSPMVDiagUpdate(IRBuilder<> &B, Module &M, BlasInfo blas,
   auto FDiagUpdateT = FunctionType::get(
       B.getVoidTy(),
       {BlasCT, BlasIT, BlasFPT, BlasPT, BlasIT, BlasPT, BlasIT, BlasPT}, false);
-  Function *F = cast<Function>(
-      M.getOrInsertFunction(fnc_name, FDiagUpdateT).getCallee());
+  Function *F =
+      cast<Function>(M.getOrInsertFunction(fnc_name, FDiagUpdateT).getCallee());
 
   if (!F->empty()) {
     B.CreateCall(F, args, bundles);

diff --git a/enzyme/tools/enzyme-tblgen/blasDiffUseUpdater.h b/enzyme/tools/enzyme-tblgen/blasDiffUseUpdater.h
@@ -83,7 +83,8 @@ void emit_BLASDiffUse(TGPattern &pattern, llvm::raw_ostream &os) {
 
   if (hasDiffeRetVal) {
     size_t ptrRetArg = typeMap.size();
-    Twine retarg = "CI->getArgOperand(" + Twine(ptrRetArg) + " + offset)";
+    auto retarg =
+        "CI->getArgOperand(" + std::to_string(ptrRetArg) + " + offset)";
     os << "  if (cublas) {\n";
     os << "    if (!gutils->isConstantValue(" << retarg << "))\n";
     os << "      if ((shadow || EnzymeRuntimeActivityCheck) && val == "