mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-21 06:16:06 +00:00
Code Issues Packages Projects Releases Wiki Activity
26,443 Commits 508 Branches 263 Tags
rocm-main
Commit Graph

4 Commits

Author SHA1 Message Date
Ilia Sergachev
e083c08001 Re-enable cudnn_fusion_test on A100. 
Check that the required cuDNN version is available.
 2024-11-01 15:48:07 +00:00
Peter Hawkins
1260ebbe05 Disable cudnn_fusion_test on A100. 
This test only seems to pass on H100 at the moment.

PiperOrigin-RevId: 681070398
 2024-10-01 10:18:41 -07:00
Ilia Sergachev
b320dc2e5e Fix and reenable cudnn_fusion_test. 
Disable XLA autotuning fallback to cuBLAS so that the tested fusion
always executes through cuDNN.
 2024-09-30 14:03:55 +00:00
Ilia Sergachev
85d792a92d Add cudnn_fusion decorator lowering computations to XLA cuDNN fusions. 2024-09-05 01:25:54 +02:00