mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-19 05:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cuda
History
Peter Hawkins a999120514 Improve error message when cudnn is not found. 
We infer a missing cudnn if cudnnGetVersion() returns 0, since the stub implementation in TSL will do that if the library isn't found (10a378f499/third_party/tsl/tsl/cuda/cudnn_stub.cc (L58)).

PiperOrigin-RevId: 587056454
2023-12-01 10:52:48 -08:00
..
BUILD
Add minimum version checks for cublas and cusparse.
2023-11-17 19:30:41 -08:00
versions_helpers.cc
Improve error message when cudnn is not found.
2023-12-01 10:52:48 -08:00
versions_helpers.h
Improve error message when cudnn is not found.
2023-12-01 10:52:48 -08:00
versions.cc
Improve error message when cudnn is not found.
2023-12-01 10:52:48 -08:00