Method with_stream

crates/blastoff/src/context.rs:128–142

Executes a given closure in a specific CUDA [`Stream`], specifically, it sets the current cublas stream for the context, runs the closure, then unsets the stream back to NULL.

(
        &mut self,
        stream: &Stream,
        func: F,
    )

    /// Executes a given closure in a specific CUDA [`Stream`], specifically, it sets the current cublas stream
    /// for the context, runs the closure, then unsets the stream back to NULL.
    pub fn with_stream<T, F: FnOnce(&mut Self) -> Result<T>>(
        &mut self,
        stream: &Stream,
        func: F,
    ) -> Result<T> {
        unsafe {
            // cudaStream_t is the same as CUstream
            sys::v2::cublasSetStream_v2(self.raw, mem::transmute(stream.as_inner())).to_result()?;
            let res = func(self)?;
            // reset the stream back to NULL just in case someone calls with_stream, then drops the stream, and tries to
            // execute a raw sys function with the context's handle.
            sys::v2::cublasSetStream_v2(self.raw, ptr::null_mut()).to_result()?;
            Ok(res)
        }
    }

    /// Sets whether the cuBLAS library is allowed to use atomics for certain routines such as `symv` or `hemv`.
    ///