NVIDIA / NVIDIA/cccl

Initial DeviceScan deferred num_items support only for look-ahead and non-deterministic

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

Description

Add initial support for device-resident num_items in cub::DeviceScan, limited to the Warpspeed look-ahead path in non-deterministic mode. The kernel will read num_items from the device-resident source on entry, derive num_tiles and its per-tile work range on device, and perform per-tile prefix propagation through the existing look-ahead global tile-state array. For temp-storage for the tile-state buffer we can use a caller-provided upper bound on num_items (via cuda::args::deferred bounds) or something like INT_MAX.

This issue can be closed once the look-ahead non-deterministic works locally with minimal testing.

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.