NVIDIA / NVIDIA/cccl

cuda.compute: implement functionality needed for Awkward (jagged) arrays

Open
#6,697 1 comment 0 reactions 1 assignee Claimed by @shwina View on GitHub
Dominant language
C++
Stars
2.5k
Forks
486
Avg merge
2d 6h
Merged PRs (30d)
295

Description

This meta issue tracks work related to supporting operations on Awkward arrays.

A starting point for this effort is the initial example in https://github.com/scikit-hep/awkward/pull/3734. While this functionality can be implemented in terms of existing `cuda.compute` primitives at the time of writing, we can optimize even further by exposing and implementing more CUB/Thrust functionality.

# Algorithms needed

## P0:

These can't really be expressed efficiently with existing algorithms exposed in `cuda.compute`:

- [x] #6695
- [x] #6688

## P1:

These can be expressed somewhat efficiently; maybe requiring more passes on the data than necessary:

- [ ] #7123
- [ ] #7125

## P2:

These are nice to haves:

- [ ] #7124
- [ ] #7126
- [ ] #7127

Contributor guide

Open the contributing guide

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.