TensorPlay
API symbolscuda
Copy
View MarkdownDownload .md

tensorplay.cuda.current_blas_handle

tensorplay.cuda.current_blas_handle()[source]

Return cublasHandle_t pointer to current cuBLAS handle

tensorplay.cuda.cudart

Retrieves the CUDA runtime API module.

tensorplay.cuda.current_device

Return the index of a currently selected device.

Ask DeepWiki