diff --git a/enzyme/Enzyme/CallDerivatives.cpp b/enzyme/Enzyme/CallDerivatives.cpp index b8154840988..a4901c1cf45 100644 --- a/enzyme/Enzyme/CallDerivatives.cpp +++ b/enzyme/Enzyme/CallDerivatives.cpp @@ -1150,7 +1150,8 @@ void AdjointGenerator::handleMPI(llvm::CallInst &call, llvm::Function *called, C = CE->getOperand(0); } if (auto GV = dyn_cast(C)) { - if (GV->getName() == "ompi_mpi_op_sum") { + if (GV->getName() == "ompi_mpi_op_sum" || + GV->getName() == "RSMPI_SUM") { isSum = true; } }