zad2 wstępnie działa

DergunPiotr-WaskoDominik/zad2/macierz_mpi.cpp

@@ -11,14 +11,11 @@ void array_alloc(long **&arr, long arrsize)
 {
 	try
 	{	
-		arr = new long* [arrsize];
-		
-		
+		arr = new long* [arrsize];	
 		arr[0] = new long [arrsize*arrsize];
 	
 		for(long i = 1; i < arrsize; ++i)
 			arr[i] = arr[i-1] + arrsize;
-			
 	}
 	catch (bad_alloc& ex)
 	{
@@ -31,8 +28,7 @@ void vect_alloc(long *&arr, long arrsize)
 {
 	try
 	{	
-		arr = new long [arrsize];
-			
+		arr = new long [arrsize];	
 	}
 	catch (bad_alloc& ex)
 	{
@@ -49,15 +45,13 @@ void array_destroy(long **arr)
 
 int main(int argc, char *argv[])
 {
-	long **A=NULL, **B=NULL, **C=NULL, *B_rot=NULL, *vect=NULL, *vect_c=NULL;
-	long rozmiar=0;
+	long **A=NULL, **B=NULL, **C=NULL, *B_rot=NULL, *vect=NULL, *vect_c=NULL, *C_rot=NULL;
+	long rozmiar=0, porcja_new=0;
 	char *endptr;
 	TTiming tt;
-	long i, j;//,k;
+	long i;
 	MPI::Status  status;
 
-	
-
 	if (argc < 2)
 	{
 		cerr << "Usage: " << argv[0] << " <size>" << endl;
@@ -78,11 +72,10 @@ int main(int argc, char *argv[])
 		exit(1);
 	}
 	  
-	  	MPI::Init(argc, argv);
+	MPI::Init(argc, argv);
 
 	int taskid = MPI::COMM_WORLD.Get_rank();
 	int ntasks = MPI::COMM_WORLD.Get_size();
-	printf("ntasks= %d : taskid= %d  : Hello World!\n",ntasks,taskid);
 
 	//alokacja macierzy
 	array_alloc(A, rozmiar);
@@ -90,25 +83,33 @@ int main(int argc, char *argv[])
 	{
 		array_alloc(B, rozmiar);
 		vect_alloc(B_rot, rozmiar*rozmiar);
+		vect_alloc(C_rot, rozmiar*rozmiar);
 	}
 
-	int porcja = (rozmiar*rozmiar)/ntasks + (rozmiar*rozmiar)%ntasks;
+	// porcja - ile wektorów dostaje jeden proces do liczenia
+	// jeżeli liczba procesów większa od wektorów - każdy proces dostaje po jednym, niektóre wcale
+	long porcja = (rozmiar>=ntasks) ? (long)round(rozmiar*1.0/ntasks) : 1;
+	porcja *= rozmiar; //porcję mnożymy przez ilość elementów w jednym wektorze
+	porcja_new = porcja;
+	
 	array_alloc(C, rozmiar);
 	vect_alloc(vect, porcja);
 	vect_alloc(vect_c, porcja);
 	if (taskid == 0)
 	{
-		//wypelnienie macierzy A liczbami "losowymi"
+		// wypełnienie macierzy A liczbami "losowymi"
 		for (long i=0; i<rozmiar; ++i)
 			for (long j=0; j<rozmiar; ++j)
-				A[i][j] = (long)(sin(i) * i * j) % 10;
+				A[i][j] = 3*i+j+1; //test
+				//A[i][j] = (long)(sin(i) * i * j) % 10;
 			
-		//wypelnienie macierzy B liczbami "losowymi"
+		// wypełnienie macierzy B liczbami "losowymi"
 		for (long i=0; i<rozmiar; ++i)
 			for (long j=0; j<rozmiar; ++j)
-				B[i][j] = (long)(cos(j) *(i+j)) % 10;
+				B[i][j] = j; //test
+				//B[i][j] = (long)(cos(j) *(i+j)) % 10;
 				
-		//dokonaj obracania macierzy
+		// dokonaj obracania macierzy
 		for (long i=0; i<rozmiar; ++i)
 			for (long j=0; j<rozmiar; ++j)
 				B_rot[i*rozmiar+j] = B[j][i];
@@ -116,7 +117,7 @@ int main(int argc, char *argv[])
 	
 	//wysyłanie macierzy A do wszystkich
 	MPI::COMM_WORLD.Bcast(&A[0][0], rozmiar*rozmiar, MPI::LONG, 0);
-	if (taskid == 0)
+	if (taskid == 0  && false)
 	{
 		cout << endl << "WEKTOR DANYCH: " << endl;
 		for (i=0; i<rozmiar*rozmiar; ++i)
@@ -125,55 +126,62 @@ int main(int argc, char *argv[])
 	}
 	MPI::COMM_WORLD.Barrier();
 	
-	//MPI::COMM_WORLD.Scatter(&B_rot[0], porcja, MPI::LONG, &vect[0], porcja, MPI::LONG, 0);
-
-	
-	// każdy proces ma policzyć łącznie "porcja" pól, 
-	// dlatego dostaje wektory kolumn, w których leżą porcje
-	long pos, pocz, kon;
-	for (i=0; i<ntasks; ++i)
-	{	
-		porcja = (i == ntasks - 1) ? rozmiar*rozmiar%porcja : porcja;
-
-		pos = i*porcja;
-		pocz = pos - (pos%rozmiar);
-		kon = pos+porcja - ((pos+porcja)%rozmiar) - 1;
-		MPI::COMM_WORLD.Isend(&B_rot[pocz], kon-pocz, MPI::LONG, i, 0);
-	} 
-	
-	// w przypadku gdy iloraz wielkości macierzy i ilości programów daje resztę
-	// wszystkie porcje zostają zwiększone, a ostatnia porcja w ostatnim programie
-	// jest mniejsza - należy policzyć tą porcję zawczasu...
+	//dla wszystkich lub "rozmiar" procesów należy zadać porcję do obliczenia
+	MPI::COMM_WORLD.Scatter(&B_rot[0], porcja, MPI::LONG, &vect[0], porcja, MPI::LONG,0);// (rozmiar>=ntasks) ? 0 : ntasks-rozmiar);
 	
+	//
+	// część licząca ---------------------------------------------------------------------------------
+	//
 	
+	if (taskid == 0)
+		tt.Begin();
+
+	if (taskid == ntasks -1) //przelicz porcję dla ostatniego procesu (w przypadku dzielenia z resztą)
+	{
+		porcja_new = (rozmiar - (porcja/rozmiar)*(ntasks-1))*rozmiar;
+		
+		// myk jest taki, że muszę podać jakąś niezerową porcję do gathera,
+		// mimo, że proces nie ma żadnych (sensownych) danych do przetworzenia
+		if (porcja_new > 0)
+			porcja = porcja_new;
+	}
+
+	for (i=0; i<porcja_new; ++i)
+	{
+		vect_c[i] = 0;
+		for (long k=0; k<rozmiar; ++k)
+		{
+			vect_c[i] += A[i%rozmiar][k]*vect[(i/rozmiar)*rozmiar+k];
+		}
+	}
+	// zwróć wszystko do programu głównego
+	MPI::COMM_WORLD.Gather(&vect_c[0], porcja, MPI::LONG, &C_rot[0], porcja, MPI::LONG, 0);
+	MPI::COMM_WORLD.Barrier();
+
+	if (taskid == 0)
+	{
+		long elapsed = tt.End();	
+		cout << "Time: " << elapsed << " ms" << endl;
+	}
 	
-	/*
-	tt.Begin();
-	for (i=0; i<rozmiar; ++i) 
-		for (j=0; j<rozmiar; ++j)
+	if (taskid == 0 && false)
+	{
+		for (i=0; i<rozmiar*rozmiar; ++i)
 		{
-			C[i][j] = 0;
-			for (long k=0; k<rozmiar; ++k)
-				C[i][j] += A[i][k]*B[k][j];
-		}	
+			cout << C_rot[i] << " ";
+		}
 		
-	long elapsed = tt.End();	
-	
-	cout << "Time: " << elapsed << " ms" << endl;
+		cout << endl;
+	}
 	
+	/*
 
 	array_destroy(A);
 	array_destroy(B);
 	array_destroy(C);
 	*/
 	
-	/*
-	cout << taskid << ": " ;
-	
-	for (i=0; i<porcja; ++i)
-		cout << vect[i] << " ";
-	cout << endl;
-	*/MPI::Finalize();
+	MPI::Finalize();
 	
 	exit(0);
 }