mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-18 21:06:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/jaxlib/cuda
History
Peter Hawkins 41f0b336e3 Add minimum version checks for cublas and cusparse. 
Split code to determine CUDA library versions out of py_extension() module and into a cc_library(), because it fixes a linking problem in Google's build. (Long story, not worth it.)

Fixes https://github.com/google/jax/issues/8289

PiperOrigin-RevId: 583544218
2023-11-17 19:30:41 -08:00
..
BUILD
Add minimum version checks for cublas and cusparse.
2023-11-17 19:30:41 -08:00
versions_helpers.cc
Add minimum version checks for cublas and cusparse.
2023-11-17 19:30:41 -08:00
versions_helpers.h
Add minimum version checks for cublas and cusparse.
2023-11-17 19:30:41 -08:00
versions.cc
Add minimum version checks for cublas and cusparse.
2023-11-17 19:30:41 -08:00