Skip to content

Commit 247111b

Browse files
jfacevedo-googlejfacevedo-google
committed
reverts accidental change that removes attn_mask in attn. Improves flux ptxla by using flash block sizes. Moves encoding outside the for loop.
1 parent 8ead643 commit 247111b

File tree

3 files changed

+133
-52
lines changed

3 files changed

+133
-52
lines changed

0 commit comments

Comments
 (0)