You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Is your feature request related to a problem? Please describe.
We should enable per thread default streams in pylibcudf.
Describe the solution you'd like
I think we can do this by exposing cudf::is_ptds_enabled in pylibcudf and using it to set the default stream to PER_THREAD_DEFAULT_STREAM from pylibrmm.
Did you see #17626? I don't know if we will ever ship builds for Python usage that have PTDS enabled at compile-time by default. If we did, though, rather than using is_ptds_enabled couldn't we just wrap get_default_stream in Python directly?
Did you see #17626? I don't know if we will ever ship builds for Python usage that have PTDS enabled at compile-time by default. If we did, though, rather than using is_ptds_enabled couldn't we just wrap get_default_stream in Python directly?
No I didn't, thanks for the link. Yes I think we could just expose get_default_stream. It would look something likecpdef Stream get_default_stream().
I assigned #17626 to me, but we can probably discuss this more offline before I start actively working on it.
Is your feature request related to a problem? Please describe.
We should enable per thread default streams in pylibcudf.
Describe the solution you'd like
I think we can do this by exposing
cudf::is_ptds_enabled
in pylibcudf and using it to set the default stream toPER_THREAD_DEFAULT_STREAM
from pylibrmm.Describe alternatives you've considered
None
Additional context
#17620
The text was updated successfully, but these errors were encountered: