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

Set decompositions for some nprocs by hand

Browse Source
This commit is contained in:
jpekkila
2020-03-30 18:13:50 +03:00
parent 9065381b2a
commit 24e65ab02d
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/core/device.cc
View File

@@ -495,6 +495,10 @@ decompose(const int target)
return (int3){4, 2, 2}; return (int3){4, 2, 2};
if (target == 32) if (target == 32)
return (int3){4, 4, 2}; return (int3){4, 4, 2};
if (target == 128)
return (int3){8, 4, 4};
if (target == 256)
return (int3){8, 8, 4};
int decomposition[] = {1, 1, 1}; int decomposition[] = {1, 1, 1};