jax
cc9f6e75 - [Pallas] Fix GQA triton kernel test.

Commit

1 year ago

[Pallas] Fix GQA triton kernel test. PiperOrigin-RevId: 715576240

References

#25888 - [Pallas] Fix GQA triton kernel test.

#213 - CI: 01/27/25 upstream sync

Author

justinjfu

justinjfu

Committer

Google-ML-Automation

Google-ML-Automation

Parents

Loading