add RSMPI_SUM

EnzymeAD · Aug 9, 2024 · 904cadf · 904cadf
1 parent 1b41f0b
commit 904cadf
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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<GlobalVariable>(C)) {
-          if (GV->getName() == "ompi_mpi_op_sum") {
+          if (GV->getName() == "ompi_mpi_op_sum" ||
+              GV->getName() == "RSMPI_SUM") {
             isSum = true;
           }
         }