zad2 wstępnie działa
This commit is contained in:
@@ -12,13 +12,10 @@ void array_alloc(long **&arr, long arrsize)
|
|||||||
try
|
try
|
||||||
{
|
{
|
||||||
arr = new long* [arrsize];
|
arr = new long* [arrsize];
|
||||||
|
|
||||||
|
|
||||||
arr[0] = new long [arrsize*arrsize];
|
arr[0] = new long [arrsize*arrsize];
|
||||||
|
|
||||||
for(long i = 1; i < arrsize; ++i)
|
for(long i = 1; i < arrsize; ++i)
|
||||||
arr[i] = arr[i-1] + arrsize;
|
arr[i] = arr[i-1] + arrsize;
|
||||||
|
|
||||||
}
|
}
|
||||||
catch (bad_alloc& ex)
|
catch (bad_alloc& ex)
|
||||||
{
|
{
|
||||||
@@ -32,7 +29,6 @@ void vect_alloc(long *&arr, long arrsize)
|
|||||||
try
|
try
|
||||||
{
|
{
|
||||||
arr = new long [arrsize];
|
arr = new long [arrsize];
|
||||||
|
|
||||||
}
|
}
|
||||||
catch (bad_alloc& ex)
|
catch (bad_alloc& ex)
|
||||||
{
|
{
|
||||||
@@ -49,15 +45,13 @@ void array_destroy(long **arr)
|
|||||||
|
|
||||||
int main(int argc, char *argv[])
|
int main(int argc, char *argv[])
|
||||||
{
|
{
|
||||||
long **A=NULL, **B=NULL, **C=NULL, *B_rot=NULL, *vect=NULL, *vect_c=NULL;
|
long **A=NULL, **B=NULL, **C=NULL, *B_rot=NULL, *vect=NULL, *vect_c=NULL, *C_rot=NULL;
|
||||||
long rozmiar=0;
|
long rozmiar=0, porcja_new=0;
|
||||||
char *endptr;
|
char *endptr;
|
||||||
TTiming tt;
|
TTiming tt;
|
||||||
long i, j;//,k;
|
long i;
|
||||||
MPI::Status status;
|
MPI::Status status;
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
if (argc < 2)
|
if (argc < 2)
|
||||||
{
|
{
|
||||||
cerr << "Usage: " << argv[0] << " <size>" << endl;
|
cerr << "Usage: " << argv[0] << " <size>" << endl;
|
||||||
@@ -78,11 +72,10 @@ int main(int argc, char *argv[])
|
|||||||
exit(1);
|
exit(1);
|
||||||
}
|
}
|
||||||
|
|
||||||
MPI::Init(argc, argv);
|
MPI::Init(argc, argv);
|
||||||
|
|
||||||
int taskid = MPI::COMM_WORLD.Get_rank();
|
int taskid = MPI::COMM_WORLD.Get_rank();
|
||||||
int ntasks = MPI::COMM_WORLD.Get_size();
|
int ntasks = MPI::COMM_WORLD.Get_size();
|
||||||
printf("ntasks= %d : taskid= %d : Hello World!\n",ntasks,taskid);
|
|
||||||
|
|
||||||
//alokacja macierzy
|
//alokacja macierzy
|
||||||
array_alloc(A, rozmiar);
|
array_alloc(A, rozmiar);
|
||||||
@@ -90,25 +83,33 @@ int main(int argc, char *argv[])
|
|||||||
{
|
{
|
||||||
array_alloc(B, rozmiar);
|
array_alloc(B, rozmiar);
|
||||||
vect_alloc(B_rot, rozmiar*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);
|
array_alloc(C, rozmiar);
|
||||||
vect_alloc(vect, porcja);
|
vect_alloc(vect, porcja);
|
||||||
vect_alloc(vect_c, porcja);
|
vect_alloc(vect_c, porcja);
|
||||||
if (taskid == 0)
|
if (taskid == 0)
|
||||||
{
|
{
|
||||||
//wypelnienie macierzy A liczbami "losowymi"
|
// wypełnienie macierzy A liczbami "losowymi"
|
||||||
for (long i=0; i<rozmiar; ++i)
|
for (long i=0; i<rozmiar; ++i)
|
||||||
for (long j=0; j<rozmiar; ++j)
|
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 i=0; i<rozmiar; ++i)
|
||||||
for (long j=0; j<rozmiar; ++j)
|
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 i=0; i<rozmiar; ++i)
|
||||||
for (long j=0; j<rozmiar; ++j)
|
for (long j=0; j<rozmiar; ++j)
|
||||||
B_rot[i*rozmiar+j] = B[j][i];
|
B_rot[i*rozmiar+j] = B[j][i];
|
||||||
@@ -116,7 +117,7 @@ int main(int argc, char *argv[])
|
|||||||
|
|
||||||
//wysyłanie macierzy A do wszystkich
|
//wysyłanie macierzy A do wszystkich
|
||||||
MPI::COMM_WORLD.Bcast(&A[0][0], rozmiar*rozmiar, MPI::LONG, 0);
|
MPI::COMM_WORLD.Bcast(&A[0][0], rozmiar*rozmiar, MPI::LONG, 0);
|
||||||
if (taskid == 0)
|
if (taskid == 0 && false)
|
||||||
{
|
{
|
||||||
cout << endl << "WEKTOR DANYCH: " << endl;
|
cout << endl << "WEKTOR DANYCH: " << endl;
|
||||||
for (i=0; i<rozmiar*rozmiar; ++i)
|
for (i=0; i<rozmiar*rozmiar; ++i)
|
||||||
@@ -125,55 +126,62 @@ int main(int argc, char *argv[])
|
|||||||
}
|
}
|
||||||
MPI::COMM_WORLD.Barrier();
|
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,
|
if (taskid == 0)
|
||||||
// dlatego dostaje wektory kolumn, w których leżą porcje
|
tt.Begin();
|
||||||
long pos, pocz, kon;
|
|
||||||
for (i=0; i<ntasks; ++i)
|
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;
|
// myk jest taki, że muszę podać jakąś niezerową porcję do gathera,
|
||||||
pocz = pos - (pos%rozmiar);
|
// mimo, że proces nie ma żadnych (sensownych) danych do przetworzenia
|
||||||
kon = pos+porcja - ((pos+porcja)%rozmiar) - 1;
|
if (porcja_new > 0)
|
||||||
MPI::COMM_WORLD.Isend(&B_rot[pocz], kon-pocz, MPI::LONG, i, 0);
|
porcja = porcja_new;
|
||||||
}
|
}
|
||||||
|
|
||||||
// w przypadku gdy iloraz wielkości macierzy i ilości programów daje resztę
|
for (i=0; i<porcja_new; ++i)
|
||||||
// wszystkie porcje zostają zwiększone, a ostatnia porcja w ostatnim programie
|
{
|
||||||
// jest mniejsza - należy policzyć tą porcję zawczasu...
|
vect_c[i] = 0;
|
||||||
|
for (long k=0; k<rozmiar; ++k)
|
||||||
|
|
||||||
|
|
||||||
/*
|
|
||||||
tt.Begin();
|
|
||||||
for (i=0; i<rozmiar; ++i)
|
|
||||||
for (j=0; j<rozmiar; ++j)
|
|
||||||
{
|
{
|
||||||
C[i][j] = 0;
|
vect_c[i] += A[i%rozmiar][k]*vect[(i/rozmiar)*rozmiar+k];
|
||||||
for (long k=0; k<rozmiar; ++k)
|
}
|
||||||
C[i][j] += A[i][k]*B[k][j];
|
}
|
||||||
|
// 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] << " ";
|
||||||
}
|
}
|
||||||
|
|
||||||
long elapsed = tt.End();
|
cout << endl;
|
||||||
|
}
|
||||||
cout << "Time: " << elapsed << " ms" << endl;
|
|
||||||
|
|
||||||
|
/*
|
||||||
|
|
||||||
array_destroy(A);
|
array_destroy(A);
|
||||||
array_destroy(B);
|
array_destroy(B);
|
||||||
array_destroy(C);
|
array_destroy(C);
|
||||||
*/
|
*/
|
||||||
|
|
||||||
/*
|
MPI::Finalize();
|
||||||
cout << taskid << ": " ;
|
|
||||||
|
|
||||||
for (i=0; i<porcja; ++i)
|
|
||||||
cout << vect[i] << " ";
|
|
||||||
cout << endl;
|
|
||||||
*/MPI::Finalize();
|
|
||||||
|
|
||||||
exit(0);
|
exit(0);
|
||||||
}
|
}
|
||||||
Reference in New Issue
Block a user