Save first 10 frontier backward matrices in mtx format.

Author: biqar

Applications/BetwCent.cpp

@@ -86,9 +86,11 @@ int main(int argc, char* argv[])
 		int K4Approx = atoi(argv[2]);
 		int batchSize = atoi(argv[3]);
 
-		string directory(argv[1]);		
-		string ifilename = "input.mtx";
-		ifilename = directory+"/"+ifilename;
+//		string directory(argv[1]);
+    string ifilename(argv[1]);
+    string frontier_dir(argv[4]);
+//		string ifilename = "input.mtx";
+//		ifilename = directory+"/"+ifilename;
 
 		shared_ptr<CommGrid> fullWorld;
 		fullWorld.reset( new CommGrid(MPI_COMM_WORLD, 0, 0) );
@@ -195,18 +197,29 @@ int main(int argc, char* argv[])
 			DenseParMat<int, double> bcu(1.0, AT.getcommgrid(), fringe.getlocalrows(), fringe.getlocalcols() );
 
 			SpParHelper::Print("Tallying...\n");
-			// BC update for all vertices except the sources
-			for(int j = bfs.size()-1; j > 0; --j)
-			{
-				Dist<double>::MPI_DCCols w = EWiseMult( *bfs[j], nspInv, false);
-				w.EWiseScale(bcu);
-
-				Dist<double>::MPI_DCCols product = PSpGEMM<PTBOOLDOUBLE>(A,w);
-				product = EWiseMult(product, *bfs[j-1], false);
-				product = EWiseMult(product, nsp, false);		
+      // BC update for all vertices except the sources
+      iter = 0;
+      for(int j = bfs.size()-1; j > 0; --j)
+      {
+        Dist<double>::MPI_DCCols w = EWiseMult( *bfs[j], nspInv, false);
+        w.EWiseScale(bcu);
+
+        if(iter < 10) {
+          string frontier_filename =
+              frontier_dir + "/" + string("batch_") + std::to_string(i) + string("_backwarditer_") +
+              std::to_string(iter) + string(".mtx");
+          w.ParallelWriteMM(frontier_filename, true);
+          iter++;
+        }
+        else {
+          exit(-1);
+        }
+        Dist<double>::MPI_DCCols product = PSpGEMM<PTBOOLDOUBLE>(A,w);
+        product = EWiseMult(product, *bfs[j-1], false);
+        product = EWiseMult(product, nsp, false);
 
-				bcu += product;
-			}
+        bcu += product;
+      }
 			for(int j=0; j < bfs.size(); ++j)
 			{
 				delete bfs[j];
@@ -226,7 +239,7 @@ int main(int argc, char* argv[])
 			fprintf(stdout, "%.6lf seconds elapsed for %d starting vertices\n", t2-t1, nPasses);
 			fprintf(stdout, "TEPS score is: %.6lf\n", TEPS);
 		}
-        ofstream output(argv[4]);
+        ofstream output(argv[5]);
         bc.SaveGathered(output, 0);
         output.close();
 	}