pioder/prir
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

dalsza zabawa z podziałami wektorów itd

Browse Source
This commit is contained in:
Piotr Dergun
2016-10-31 17:40:31 +01:00
parent 1bf0ef0dfe
commit 516768a97a
1 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
DergunPiotr-WaskoDominik/zad2/macierz_mpi.cpp
View File

@@ -27,6 +27,20 @@ void array_alloc(long **&arr, long arrsize)
} }
} }
void vect_alloc(long *&arr, long arrsize)
{
try
{
arr = new long [arrsize];
}
catch (bad_alloc& ex)
{
cerr << "Could not allocate memory for vector" << endl;
exit(1);
}
}
void array_destroy(long **arr) void array_destroy(long **arr)
{ {
delete [] arr[0]; delete [] arr[0];
@@ -35,7 +49,7 @@ 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; long **A=NULL, **B=NULL, **C=NULL, *B_rot=NULL, *vect=NULL;
long rozmiar=0; long rozmiar=0;
char *endptr; char *endptr;
TTiming tt; TTiming tt;
@@ -75,10 +89,10 @@ int main(int argc, char *argv[])
if (taskid == 0) if (taskid == 0)
{ {
array_alloc(B, rozmiar); array_alloc(B, rozmiar);
array_alloc(B_rot, rozmiar); vect_alloc(B_rot, rozmiar*rozmiar);
} }
array_alloc(C, rozmiar); array_alloc(C, rozmiar);
vect_alloc(vect, rozmiar);
if (taskid == 0) if (taskid == 0)
{ {
//wypelnienie macierzy A liczbami "losowymi" //wypelnienie macierzy A liczbami "losowymi"
@@ -94,13 +108,15 @@ int main(int argc, char *argv[])
//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][j] = B[j][i]; B_rot[i*rozmiar+j] = B[j][i];
} }
//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);
MPI::COMM_WORLD.Scatter(&B_rot[0], rozmiar, MPI::LONG, &vect[0], rozmiar, MPI::LONG, 0);
/* /*
tt.Begin(); tt.Begin();
for (i=0; i<rozmiar; ++i) for (i=0; i<rozmiar; ++i)
@@ -121,7 +137,7 @@ int main(int argc, char *argv[])
array_destroy(C); array_destroy(C);
*/ */
cout << taskid << ": " << A[2][2] << endl; cout << taskid << ": " << vect[2] << endl;
MPI::Finalize(); MPI::Finalize();
exit(0); exit(0);