Actual source code: ex3.cxx
1: static char help[] = "AO test contributed by Sebastian Steiger <steiger@purdue.edu>, March 2011\n\n";
3: /*
4: Example of usage:
5: mpiexec -n 12 ./ex3
6: mpiexec -n 30 ./ex3 -ao_type basic
7: */
9: #include <iostream>
10: #include <fstream>
11: #include <vector>
12: #include <petscvec.h>
13: #include <petscao.h>
15: using namespace std;
17: int main(int argc, char **argv)
18: {
19: AO ao;
20: IS isapp;
21: char infile[PETSC_MAX_PATH_LEN], datafiles[PETSC_MAX_PATH_LEN];
22: PetscBool flg;
23: PetscMPIInt size, rank;
25: PetscFunctionBeginUser;
26: PetscCall(PetscInitialize(&argc, &argv, nullptr, help));
27: PetscCallMPI(MPI_Comm_size(PETSC_COMM_WORLD, &size));
28: PetscCallMPI(MPI_Comm_rank(PETSC_COMM_WORLD, &rank));
30: PetscCall(PetscOptionsGetString(NULL, NULL, "-datafiles", datafiles, sizeof(datafiles), &flg));
31: PetscCheck(flg, PETSC_COMM_WORLD, PETSC_ERR_USER, "Must specify -datafiles ${DATAFILESPATH}/ao");
33: // read in application indices
34: PetscCall(PetscSNPrintf(infile, sizeof(infile), "%s/AO%dCPUs/ao_p%d_appindices.txt", datafiles, size, rank));
35: ifstream fin(infile);
36: PetscCheck(fin, PETSC_COMM_SELF, PETSC_ERR_FILE_OPEN, "File not found: %s", infile);
37: vector<PetscInt> myapp;
38: int tmp = -1;
39: while (!fin.eof()) {
40: tmp = -1;
41: fin >> tmp;
42: if (tmp == -1) break;
43: myapp.push_back(tmp);
44: }
45: #if __cplusplus >= 201103L // c++11
46: static_assert(is_same<decltype(myapp.size()), size_t>::value, "");
47: #endif
48: PetscCall(PetscSynchronizedPrintf(PETSC_COMM_WORLD, "[%d] has %zu indices.\n", rank, myapp.size()));
49: PetscCall(PetscSynchronizedFlush(PETSC_COMM_WORLD, PETSC_STDOUT));
51: PetscCall(ISCreateGeneral(PETSC_COMM_WORLD, myapp.size(), &myapp[0], PETSC_USE_POINTER, &isapp));
53: PetscCall(AOCreate(PETSC_COMM_WORLD, &ao));
54: PetscCall(AOSetIS(ao, isapp, NULL));
55: PetscCall(AOSetType(ao, AOMEMORYSCALABLE));
56: PetscCall(AOSetFromOptions(ao));
58: if (rank == 0) cout << "AO has been set up." << endl;
60: PetscCall(AODestroy(&ao));
61: PetscCall(ISDestroy(&isapp));
63: if (rank == 0) cout << "AO is done." << endl;
65: PetscCall(PetscFinalize());
66: return 0;
67: }
69: /*TEST
71: build:
72: requires: !defined(PETSC_USE_64BIT_INDICES)
74: test:
75: nsize: 12
76: requires: double !complex datafilespath
77: args: -datafiles ${DATAFILESPATH}/ao
78: output_file: output/ex3_1.out
80: test:
81: suffix: 2
82: nsize: 12
83: requires: double !complex datafilespath
84: args: -ao_type basic -datafiles ${DATAFILESPATH}/ao
85: output_file: output/ex3_1.out
87: test:
88: suffix: 3
89: nsize: 30
90: requires: double !complex datafilespath
91: args: -datafiles ${DATAFILESPATH}/ao
93: test:
94: suffix: 4
95: nsize: 30
96: requires: double !complex datafilespath
97: args: -ao_type basic -datafiles ${DATAFILESPATH}/ao
98: output_file: output/ex3_3.out
100: TEST*/