zad2 wstępnie działa
This commit is contained in:
@@ -12,13 +12,10 @@ void array_alloc(long **&arr, long arrsize)
|
||||
try
|
||||
{
|
||||
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)
|
||||
{
|
||||
@@ -32,7 +29,6 @@ void vect_alloc(long *&arr, long arrsize)
|
||||
try
|
||||
{
|
||||
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;
|
||||
@@ -82,7 +76,6 @@ int main(int argc, char *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,23 +83,31 @@ 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
|
||||
for (long i=0; i<rozmiar; ++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);
|
||||
//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 ---------------------------------------------------------------------------------
|
||||
//
|
||||
|
||||
// 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)
|
||||
if (taskid == 0)
|
||||
tt.Begin();
|
||||
|
||||
if (taskid == ntasks -1) //przelicz porcję dla ostatniego procesu (w przypadku dzielenia z resztą)
|
||||
{
|
||||
porcja = (i == ntasks - 1) ? rozmiar*rozmiar%porcja : porcja;
|
||||
porcja_new = (rozmiar - (porcja/rozmiar)*(ntasks-1))*rozmiar;
|
||||
|
||||
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);
|
||||
// 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;
|
||||
}
|
||||
|
||||
// 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...
|
||||
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;
|
||||
}
|
||||
|
||||
if (taskid == 0 && false)
|
||||
{
|
||||
for (i=0; i<rozmiar*rozmiar; ++i)
|
||||
{
|
||||
cout << C_rot[i] << " ";
|
||||
}
|
||||
|
||||
cout << endl;
|
||||
}
|
||||
|
||||
/*
|
||||
tt.Begin();
|
||||
for (i=0; i<rozmiar; ++i)
|
||||
for (j=0; j<rozmiar; ++j)
|
||||
{
|
||||
C[i][j] = 0;
|
||||
for (long k=0; k<rozmiar; ++k)
|
||||
C[i][j] += A[i][k]*B[k][j];
|
||||
}
|
||||
|
||||
long elapsed = tt.End();
|
||||
|
||||
cout << "Time: " << elapsed << " ms" << 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);
|
||||
}
|
||||
Reference in New Issue
Block a user