add n_ranks print

usainzg · Oct 29, 2024 · a104c83 · a104c83
1 parent a66af10
commit a104c83
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/lib/Transform/Affine/AffineDistributeToMPI.cpp b/lib/Transform/Affine/AffineDistributeToMPI.cpp
@@ -22,6 +22,10 @@ struct AffineDistributeToMPI
 
   // TODO: change to work with funcOp instead of affineForOp?
   void runOnOperation() {
+    // print number of ranks
+    llvm::errs() << "n_ranks=" << n_ranks << "\n";
+
+    // capture affineForOp and walk the IR
     getOperation()->walk([&](AffineForOp op) {
       OpBuilder builder(op.getContext());