std :: ml::gpu_cuda
Function std::cuda_stream_handle
?&CudaNativeStream cuda_stream_handle()The CUstream every op above is queued on (a cudaStream_t to the runtime API, cuDNN and cuFFT); null if there is no GPU. Work another library queues on it is ordered with this backend's ops and recorded by graph capture.