make sure it's a cupy zero dim array (#3271)

scikit-hep · Oct 9, 2024 · 0a92539 · 0a92539
1 parent e8f5f9a
commit 0a92539
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/src/awkward/_connect/cuda/cuda_kernels/awkward_ListArray_getitem_next_at.cu b/src/awkward/_connect/cuda/cuda_kernels/awkward_ListArray_getitem_next_at.cu
@@ -1,5 +1,12 @@
 // BSD 3-Clause License; see https://github.com/scikit-hep/awkward-1.0/blob/main/LICENSE
 
+// BEGIN PYTHON
+// def f(grid, block, args):
+// (tocarry, fromstarts, fromstops, lenstarts, at, invocation_index, err_code) = args
+// cuda_kernel_templates.get_function(fetch_specialization(["awkward_ListArray_getitem_next_at", tocarry.dtype, fromstarts.dtype, fromstops.dtype]))(grid, block, (tocarry, fromstarts, fromstops, lenstarts, cupy.array(at), invocation_index, err_code))
+// out["awkward_ListArray_getitem_next_at", {dtype_specializations}] = None
+// END PYTHON
+
 enum class LISTARRAY_GETITEM_NEXT_AT_ERRORS {
  IND_OUT_OF_RANGE, // message: "index out of range"
 };

diff --git a/src/awkward/_connect/cuda/cuda_kernels/awkward_RegularArray_getitem_next_at.cu b/src/awkward/_connect/cuda/cuda_kernels/awkward_RegularArray_getitem_next_at.cu
@@ -1,5 +1,13 @@
 // BSD 3-Clause License; see https://github.com/scikit-hep/awkward-1.0/blob/main/LICENSE
 
+
+// BEGIN PYTHON
+// def f(grid, block, args):
+// (tocarry, at, length, size, invocation_index, err_code) = args
+// cuda_kernel_templates.get_function(fetch_specialization(["awkward_RegularArray_getitem_next_at", tocarry.dtype]))(grid, block, (tocarry, cupy.array(at), length, size, invocation_index, err_code))
+// out["awkward_RegularArray_getitem_next_at", {dtype_specializations}] = None
+// END PYTHON
+
 enum class REGULARARRAY_GETITEM_NEXT_AT_ERRORS {
  IND_OUT_OF_RANGE // message: "index out of range"
 };