Function cudarc::cudnn::result::set_stream

source · 
pub unsafe fn set_stream(
    handle: cudnnHandle_t,
    stream: cudaStream_t
) -> Result<(), CudnnError>
Expand description

Sets the stream cuDNN will use. See nvidia docs

Safety

handle and stream must be valid.