jax
jax copied to clipboard
Fix and reenable cudnn_fusion_test.
Disable XLA autotuning fallback to cuBLAS so that the tested fusion always executes through cuDNN.