torch.cuda.current_blas_handle#
- torch.cuda.current_blas_handle()[source]#
Return the
cublasHandle_tpointer for the current device and stream.On CUDA, the handle uses cuBLAS’s default workspace unless ATen workspace caching is explicitly enabled. When caching is disabled, internal ATen operations may temporarily bind their own workspace, but restore the default workspace before releasing it. ROCm caches workspaces by default.