feat: turn on CUDA unit tests for working kernels and add some CUDA kernels (#2930)

* feat: add more cuda kernels

* fix: parsed python code in CUDA kernels

* tests: turn on the tests for correctly working CUDA kernels

* feat: add awkward_IndexedArray_index_of_nulls CUDA kernel

* refactor: fix indentation

* feat: add awkward_ListArray_min_range CUDA kernel

* test: add unit tests awkward_ListArray_getitem_next_range_spreadadvanced CUDA kernel

Author: ManasviGoyal

awkward-cpp/src/cpu-kernels/awkward_ListOffsetArray_drop_none_indexes.cpp

@@ -41,7 +41,6 @@ ERROR awkward_ListOffsetArray_drop_none_indexes_64(
     length_offsets,
     length_indexes);
 }
-
 ERROR awkward_ListOffsetArray_drop_none_indexes_32(
   int32_t* tooffsets,
   const int32_t* noneindexes,

dev/generate-kernel-signatures.py

@@ -12,6 +12,7 @@
 
 
 cuda_kernels_impl = [
+    "awkward_ListArray_min_range",
     "awkward_ListArray_validity",
     "awkward_BitMaskedArray_to_ByteMaskedArray",
     "awkward_ListArray_compact_offsets",
@@ -45,7 +46,9 @@
     "awkward_missing_repeat",
     "awkward_RegularArray_getitem_jagged_expand",
     "awkward_ListArray_getitem_jagged_expand",
+    "awkward_ListArray_getitem_next_array_advanced",
     "awkward_ListArray_getitem_next_array",
+    "awkward_ListArray_getitem_next_at",
     "awkward_NumpyArray_reduce_adjust_starts_64",
     "awkward_NumpyArray_reduce_adjust_starts_shifts_64",
     "awkward_RegularArray_getitem_next_at",
@@ -61,6 +64,7 @@
     "awkward_IndexedArray_flatten_nextcarry",
     "awkward_IndexedArray_getitem_nextcarry",
     "awkward_IndexedArray_getitem_nextcarry_outindex",
+    "awkward_IndexedArray_index_of_nulls",
     "awkward_IndexedArray_reduce_next_64",
     "awkward_IndexedArray_reduce_next_nonlocal_nextshifts_64",
     "awkward_IndexedArray_reduce_next_nonlocal_nextshifts_fromshifts_64",

dev/generate-tests.py

@@ -643,6 +643,7 @@ def gencpuunittests(specdict):
 
 
 cuda_kernels_tests = [
+    "awkward_ListArray_min_range",
     "awkward_ListArray_validity",
     "awkward_BitMaskedArray_to_ByteMaskedArray",
     "awkward_ListArray_compact_offsets",
@@ -676,7 +677,9 @@ def gencpuunittests(specdict):
     "awkward_missing_repeat",
     "awkward_RegularArray_getitem_jagged_expand",
     "awkward_ListArray_getitem_jagged_expand",
+    "awkward_ListArray_getitem_next_array_advanced",
     "awkward_ListArray_getitem_next_array",
+    "awkward_ListArray_getitem_next_at",
     "awkward_NumpyArray_reduce_adjust_starts_64",
     "awkward_NumpyArray_reduce_adjust_starts_shifts_64",
     "awkward_RegularArray_getitem_next_at",
@@ -692,6 +695,7 @@ def gencpuunittests(specdict):
     "awkward_IndexedArray_flatten_nextcarry",
     "awkward_IndexedArray_getitem_nextcarry",
     "awkward_IndexedArray_getitem_nextcarry_outindex",
+    "awkward_IndexedArray_index_of_nulls",
     "awkward_IndexedArray_reduce_next_64",
     "awkward_IndexedArray_reduce_next_nonlocal_nextshifts_64",
     "awkward_IndexedArray_reduce_next_nonlocal_nextshifts_fromshifts_64",