Save first 10 frontier 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) );
@@ -175,16 +177,26 @@ int main(int argc, char* argv[])
 			Dist<int>::MPI_DCCols  nsp(nsploc, AT.getcommgrid());			
 			vector < Dist<bool>::MPI_DCCols * > bfs;		// internally keeps track of depth
 
-			SpParHelper::Print("Exploring via BFS...\n");
-			while( fringe.getnnz() > 0 )
-			{
-				nsp += fringe;
-				Dist<bool>::MPI_DCCols * level = new Dist<bool>::MPI_DCCols( fringe ); 
-				bfs.push_back(level);
+      SpParHelper::Print("Exploring via BFS...\n");
+      int iter = 0;
+      while( fringe.getnnz() > 0 )
+      {
+        nsp += fringe;
+        Dist<bool>::MPI_DCCols * level = new Dist<bool>::MPI_DCCols( fringe );
+        bfs.push_back(level);
+
+        if(iter < 10) {
+          string frontier_filename = frontier_dir + "/" + string("batch_") + std::to_string(i) + string("_forwarditer_") + std::to_string(iter)+string(".mtx");
+          fringe.ParallelWriteMM(frontier_filename, true);
+          iter++;
+        }
+        else {
+          exit(-1);
+        }
 
-				fringe = PSpGEMM<PTBOOLINT>(AT, fringe);
-				fringe = EWiseMult(fringe, nsp, true);
-			}
+        fringe = PSpGEMM<PTBOOLINT>(AT, fringe);
+        fringe = EWiseMult(fringe, nsp, true);
+      }
 
 			// Apply the unary function 1/x to every element in the matrix
 			// 1/x works because no explicit zeros are stored in the sparse matrix nsp
@@ -226,7 +238,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();
 	}