Commit 73cc7a5

Google-ML-Automation

committed

Merge pull request #22670 from wenscarl:fp8_sdpa

PiperOrigin-RevId: 706693044

2 parents e655358 + c67b651 commit 73cc7a5Copy full SHA for 73cc7a5

3 files changed

+1015

-119

lines changed

jax
- _src/cudnn
  - fused_attention_stablehlo.py
- experimental/jax2tf/tests
  - primitives_test.py
tests
- fused_attention_stablehlo_test.py

3 files changed

+1015

-119

lines changed

Comments

(0)