jax
e4f3b3ff - Merge pull request #21169 from justinjfu/splash_precision_fix

Commit

1 year ago

Merge pull request #21169 from justinjfu/splash_precision_fix Disable bfloat16 on long seq lengths for splash attention kernel test

References

Author

justinjfu

Parents