Commit 388cef5

maxtext authors

committed

Use forked splash_attention_kernel

PiperOrigin-RevId: 773860482

1 parent eac885e commit 388cef5Copy full SHA for 388cef5

2 files changed

+2526

-3

lines changed

MaxText
- kernels
  - splash_attention_kernel.py
- layers
  - attentions.py

2 files changed

+2526

-3

lines changed

Comments

(0)