True if cuBLAS was built in and a CUDA device is present.
()
| 243 | |
| 244 | |
| 245 | def has_cublas() -> bool: |
| 246 | """True if cuBLAS was built in and a CUDA device is present.""" |
| 247 | return has_cuda() and _build_flag_enabled("USE_CUBLAS") |
| 248 | |
| 249 | |
| 250 | def has_nccl() -> bool: |
nothing calls this directly
no test coverage detected
searching dependent graphs…