You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: Libraries/MPI/jacobian_solver/src/02_jacobian_device_mpi_one-sided_gpu_aware/mpi3_onesided_jacobian_gpu_openmp.c
+3-3Lines changed: 3 additions & 3 deletions
Original file line number
Diff line number
Diff line change
@@ -64,7 +64,7 @@ int main(int argc, char *argv[])
64
64
*
65
65
* NOTE: For simplification and unification across samples we use single team
66
66
* to avoid extra syncronization across teams in the future */
67
-
#pragma omp target teams distribute parallel for simd is_device_ptr(in, out) num_teams(1)
67
+
#pragma omp target teams distribute parallel for is_device_ptr(in, out) num_teams(1)
68
68
/* Calculate values on borders to initiate communications early */
69
69
for (intcolumn=0; column<my_subarray.x_size; ++column) {
70
70
RECALCULATE_POINT(out, in, column, 0, row_size);
@@ -87,7 +87,7 @@ int main(int argc, char *argv[])
87
87
}
88
88
89
89
/* Offload compute loop to the device */
90
-
#pragma omp target teams distribute parallel for simd is_device_ptr(in, out) collapse(2) num_teams(1)
90
+
#pragma omp target teams distribute parallel for is_device_ptr(in, out) collapse(2) num_teams(1)
91
91
/* Recalculate internal points in parallel with communication */
92
92
for (introw=1; row<my_subarray.y_size-1; ++row) {
93
93
for (intcolumn=0; column<my_subarray.x_size; ++column) {
@@ -104,7 +104,7 @@ int main(int argc, char *argv[])
104
104
doubleresult_norm=0.0;
105
105
doublenorm=0.0;
106
106
107
-
#pragma omp target teams distribute parallel for simd is_device_ptr(b1, b2) reduction(+:norm) collapse(2)
107
+
#pragma omp target teams distribute parallel for is_device_ptr(b1, b2) reduction(+:norm) collapse(2)
108
108
for (introw=0; row<my_subarray.y_size; ++row) {
109
109
for (intcolumn=0; column<my_subarray.x_size; ++column) {
0 commit comments