Set decompositions for some nprocs by hand

2020-03-30 18:13:50 +03:00
parent 9065381b2a
commit 24e65ab02d
1 changed files with 4 additions and 0 deletions
--- a/src/core/device.cc
+++ b/src/core/device.cc
@@ -495,6 +495,10 @@ decompose(const int target)
        return (int3){4, 2, 2};
    if (target == 32)
        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};