NVIDIA / NVIDIA/cuCollections

[ENHANCEMENT]: Update build-time check since the trunk CCCL allows 1B/2B `atomic_ref`

Open
#554 2 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

P1: Should have type: improvement
Dominant language
Cuda
Stars
667
Forks
120
Avg merge
7d 5h
Merged PRs (30d)
4

Description

Is your feature request related to a problem? Please describe.

https://godbolt.org/z/a78x4qrha

Originally posted by @sleeepyjack https://github.com/NVIDIA/cuCollections/pull/549#discussion_r1684974753

Now cuda::atomic_ref can work with 1-byte and 2-byte variables so we need to update the static asserts accordingly in cuco data structures, e.g.

https://github.com/NVIDIA/cuCollections/blob/a7f87ac838320fe5b9fc9a90847d3f652ccec201/include/cuco/static_map_ref.cuh#L77-L78

https://github.com/NVIDIA/cuCollections/blob/e7b5a389b823bebe465cefe63ad7b0e95f7fb450/include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh#L1239

Describe the solution you'd like

Update build checks and unit tests

Describe alternatives you've considered

No response

Additional context

No response

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the static assertions in include/cuco/static_map_ref.cuh and include/cuco/detail/open_addressing/open_addressing_ref_impl.cuh, then inspect the existing build checks and unit tests. Update the checks for 1-byte and 2-byte atomic_ref variables and adjust the tests accordingly. Done means the build checks and unit tests pass with the trunk CCCL behavior.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp
Domain
data
Issue type
Feature
Difficulty
3/5
Estimated time
1-2 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
48/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.