Increase the maximum number of kernel arguments to 128 (#709)

libocca · Sep 19, 2023 · ca23135 · ca23135
1 parent 48f1d63
commit ca23135
Show file tree

Hide file tree

Showing 6 changed files with 5,815 additions and 2 deletions.
diff --git a/include/occa/core/kernelOperators.hpp_codegen b/include/occa/core/kernelOperators.hpp_codegen
diff --git a/include/occa/defines/macros.hpp b/include/occa/defines/macros.hpp
@@ -26,6 +26,13 @@
 
 #  define OCCA_ARG_COUNT(...) OCCA_ARG_COUNT2(\
     __VA_ARGS__,                              \
+    128, 127, 126, 125, 124, 123, 122, 121,  \
+    120, 119, 118, 117, 116, 115, 114, 113, 112, 111,  \
+    110, 109, 108, 107, 106, 105, 104, 103, 102, 101,  \
+    100, 99, 98, 97, 96, 95, 94, 93, 92, 91,  \
+    90, 89, 88, 87, 86, 85, 84, 83, 82, 81,   \
+    80, 79, 78, 77, 76, 75, 74, 73, 72, 71,   \
+    70, 69, 68, 67, 66, 65, 64, 63, 62, 61,   \
     60, 59, 58, 57, 56, 55, 54, 53, 52, 51,   \
     50, 49, 48, 47, 46, 45, 44, 43, 42, 41,   \
     40, 39, 38, 37, 36, 35, 34, 33, 32, 31,   \
@@ -41,6 +48,13 @@
   _31, _32, _33, _34, _35, _36, _37, _38, _39, _40, \
   _41, _42, _43, _44, _45, _46, _47, _48, _49, _50, \
   _51, _52, _53, _54, _55, _56, _57, _58, _59, _60, \
+  _61, _62, _63, _64, _65, _66, _67, _68, _69, _70, \
+  _71, _72, _73, _74, _75, _76, _77, _78, _79, _80, \
+  _81, _82, _83, _84, _85, _86, _87, _88, _89, _90, \
+  _91, _92, _93, _94, _95, _96, _97, _98, _99, _100, \
+  _101, _102, _103, _104, _105, _106, _107, _108, _109, _110, \
+  _111, _112, _113, _114, _115, _116, _117, _118, _119, _120, \
+  _121, _122, _123, _124, _125, _126, _127, _128, \
   N,  ...) N
 
 #endif // OCCA_DISABLE_VARIADIC_MACROS

diff --git a/include/occa/defines/occa.hpp b/include/occa/defines/occa.hpp
@@ -15,7 +15,7 @@
 #define OKL_VERSION       10600
 #define OKL_VERSION_STR   "1.6.0"
 
-#define OCCA_MAX_ARGS 60
+#define OCCA_MAX_ARGS 128
 
 #define OCCA_DEFAULT_MEM_BYTE_ALIGN 32
 

diff --git a/scripts/codegen/setup_kernel_operators.py b/scripts/codegen/setup_kernel_operators.py
@@ -24,7 +24,7 @@
 // =========================================
 '''.strip()
 
-MAX_ARGS = 60
+MAX_ARGS = 128
 
 
 def to_file(filename):