Merge pull request #1247 from ExtremeFLOW/fix/axhelm_cuda_padding

Add missing call to padded version of Axhelm
ExtremeFLOW · Apr 22, 2024 · bf4099e · bf4099e
2 parents 7897031 + fc62f28
commit bf4099e
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/math/bcknd/device/cuda/ax_helm.cu b/src/math/bcknd/device/cuda/ax_helm.cu
@@ -137,16 +137,21 @@ extern "C" {
       CASE_KSTEP(LX);                                                           \
       break
 
+#define CASE_LARGE_PADDED(LX)                                                   \
+    case LX:                                                                    \
+      CASE_KSTEP_PADDED(LX);                                                    \
+      break
+
 
     if ((*lx) < 12) {      
       switch(*lx) {
         CASE(2);
         CASE(3);
-        CASE(4);
+        CASE_PADDED(4);
         CASE(5);
         CASE(6);
         CASE(7);
-        CASE(8);
+        CASE_PADDED(8);
         CASE(9);
         CASE(10);
         CASE(11);
@@ -163,7 +168,7 @@ extern "C" {
         CASE_LARGE(13);
         CASE_LARGE(14);
         CASE_LARGE(15);
-        CASE_LARGE(16);
+        CASE_LARGE_PADDED(16);
       default:
         {
           fprintf(stderr, __FILE__ ": size not supported: %d\n", *lx);