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 == "