cuCollections icon indicating copy to clipboard operation
cuCollections copied to clipboard

[FEA] CUDA stream support

Open seunghwak opened this issue 5 years ago • 1 comments

Is your feature request related to a problem? Please describe. static_map/dynamic_map currently does not take cudaStream_t stream parameter. This often requires additional synchronization and limits speedup we can get when we wish to concurrently run multiple cuCollection operations using multiple CUDA streams.

Describe the solution you'd like Add CUDA stream support

Additional context cuGraph needs this to run multiple graph kernels concurrently using multiple CUDA streams (for batch processing).

seunghwak avatar Nov 16 '20 22:11 seunghwak

Stream support for static_map added in commit https://github.com/NVIDIA/cuCollections/commit/76a6ba1d08761f7e14f27c6794ca2d469301b54f

chirayuG-nvidia avatar Dec 20 '21 20:12 chirayuG-nvidia