True if cuDNN was built in and a CUDA device is present.
()
| 238 | |
| 239 | |
| 240 | def has_cudnn() -> bool: |
| 241 | """True if cuDNN was built in and a CUDA device is present.""" |
| 242 | return has_cuda() and _build_flag_enabled("USE_CUDNN") |
| 243 | |
| 244 | |
| 245 | def has_cublas() -> bool: |
nothing calls this directly
no test coverage detected
searching dependent graphs…