cwpearson/astaroth
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Implemented acIntegrateStepWithOffset

Browse Source
This commit is contained in:
jpekkila
2019-07-04 16:31:16 +03:00
parent a53e0a170d
commit 5049dadc1c
1 changed files with 13 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
src/core/astaroth.cu
View File

@@ -266,14 +266,19 @@ acIntegrateStep(const int& isubstep, const AcReal& dt)
AcResult
acIntegrateStepWithOffset(const int& isubstep, const AcReal& dt, const int3& start, const int3& end)
{
/*
// A skeleton function for computing integrations with arbitrary subblocks
// Uncommenting the following should work with a single GPU.
const int3 start = (int3){NGHOST, NGHOST, NGHOST};
const int3 end = (int3){NGHOST + subgrid.n.x, NGHOST + subgrid.n.y,
NGHOST + subgrid.n.z};
*/
rkStep(devices[0], STREAM_PRIMARY, isubstep, start, end, dt);
WARNING("acIntegrateStepWithOffset called. This function has not been tested for correctness!");
for (int i = 0; i < num_devices; ++i) {
// DECOMPOSITION OFFSET HERE
const int3 d0 = (int3){NGHOST, NGHOST, NGHOST + i * subgrid.n.z};
const int3 d1 = d0 + (int3){subgrid.n.x, subgrid.n.y, subgrid.n.z};
const int3 da = max(start, d0);
const int3 db = min(end, d1);
if (db.z >= da.z) {
rkStep(devices[i], STREAM_PRIMARY, isubstep, da, db, dt);
}
}
return AC_SUCCESS;
}