rmm icon indicating copy to clipboard operation
rmm copied to clipboard

[FEA] Improvements to RMM stream pool

Open harrism opened this issue 2 years ago • 0 comments

Is your feature request related to a problem? Please describe. Extend RMM stream_pool with the same features as were recently added to libcudf in https://github.com/rapidsai/cudf/pull/13922

  • add a base stream_pool interface class
  • implement the interface using a pool like the current one
  • Also implement a default_stream_pool that always returns the default stream (for debugging)
  • Add fork / join functions for forking a set of streams from a specified stream (using the pool to supply streams), and to join a list of streams to another stream.

Describe the solution you'd like https://github.com/rapidsai/cudf/pull/13922

harrism avatar Sep 13 '23 01:09 harrism