From fc62f28ee0e2df861579e7fd5e0ad2010968a53b Mon Sep 17 00:00:00 2001
From: Niclas Jansson <njansson@kth.se>
Date: Mon, 22 Apr 2024 10:25:54 +0200
Subject: [PATCH] Add missing call to padded version of Axhelm

---
 src/math/bcknd/device/cuda/ax_helm.cu | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/src/math/bcknd/device/cuda/ax_helm.cu b/src/math/bcknd/device/cuda/ax_helm.cu
index 4872ddcef9b2..58f018249544 100644
--- 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);