Preparations for multi-GPU optimizations

This commit is contained in:
jpekkila
2019-07-05 15:44:30 +03:00
parent ce8fe53f91
commit 2092adc0f6

View File

@@ -386,7 +386,7 @@ acSynchronize(void)
AcResult AcResult
acBoundcondStep(void) acBoundcondStep(void)
{ {
acSynchronize(); acSynchronizeStream(STREAM_ALL);
if (num_devices == 1) { if (num_devices == 1) {
boundcondStep(devices[0], STREAM_PRIMARY, (int3){0, 0, 0}, subgrid.m); boundcondStep(devices[0], STREAM_PRIMARY, (int3){0, 0, 0}, subgrid.m);
} }
@@ -445,6 +445,8 @@ acBoundcondStep(void)
<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<< <<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<<
*/ */
} }
// TODO, better to use acSynchronizeStream here and let the user call acSynchronizeHalos when
// needed
acSynchronize(); acSynchronize();
return AC_SUCCESS; return AC_SUCCESS;
} }