Commit ea8776a

maxtext authors

committed

Use forked splash_attention_kernel

PiperOrigin-RevId: 773860482

1 parent 028bc3c commit ea8776aCopy full SHA for ea8776a

2 files changed

+2524

-1

lines changed

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

2 files changed

+2524

-1

lines changed

Comments

(0)