remaining CUDA kernels not yet in main - Githubissues

scikit-hep / awkward

Manipulate JSON-like data with NumPy-like idioms.

https://awkward-array.org

BSD 3-Clause "New" or "Revised" License

807 stars 81 forks source link

remaining CUDA kernels not yet in main #3173

Open jpivarski opened 6 days ago

jpivarski commented 6 days ago

The following are in development in #3150:

[x] awkward_ListArray_combinations
[x] awkward_RegularArray_combinations_64

The following are in development in #3149:

[ ] awkward_reduce_argmax_complex
[ ] awkward_reduce_argmin_complex
[x] awkward_reduce_countnonzero_complex
[x] awkward_reduce_max_complex
[x] awkward_reduce_min_complex
[x] awkward_reduce_prod_bool_complex
[ ] awkward_reduce_prod_complex
[x] awkward_reduce_sum_bool_complex
[x] awkward_reduce_sum_complex

The following are for axis=-1 reduction (I thought these were done):

[x] awkward_IndexedArray_local_preparenext_64

The following are for general reduction:

[ ] awkward_ListOffsetArray_reduce_nonlocal_nextshifts_64
[ ] awkward_ListOffsetArray_reduce_nonlocal_outstartsstops_64
[ ] awkward_ListOffsetArray_reduce_nonlocal_preparenext_64
[ ] awkward_RecordArray_reduce_nonlocal_outoffsets_64

The following are for sorting:

[ ] awkward_ListOffsetArray_argsort_strings
[ ] awkward_NumpyArray_sort_asstrings_uint8
[ ] awkward_argsort
[ ] awkward_sort

The following are for uniqueness checking/computing:

[ ] awkward_IndexedArray_unique_next_index_and_offsets_64
[ ] awkward_NumpyArray_unique_strings
[ ] awkward_unique_offsets
[ ] awkward_unique_ranges
[ ] awkward_unique_ranges_bool

The following are for converting strings to_numpy:

[ ] awkward_NumpyArray_prepare_utf8_to_utf32_padded
[ ] awkward_NumpyArray_utf8_to_utf32_padded

jpivarski commented 3 days ago

@lgray requests the sorting kernels to be next.