默认流#

group default_stream

函数

rmm::cuda_stream_view const get_default_stream()#

获取当前的默认流。

Returns:

当前的默认流。

bool is_ptds_enabled()#

检查是否启用了每线程默认流。

Returns:

如果启用了PTDS，则为true，否则为false。