squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
395e5a0dba
flash-attention/tests/losses
History
Tri Dao 713bd3aa9a [CrossEntropy] Test longer sequences
2023-12-16 19:11:23 -08:00
..
test_cross_entropy_parallel.py [CrossEntropy] Implement logit_scale option 2023-12-16 18:39:37 -08:00
test_cross_entropy.py [CrossEntropy] Test longer sequences 2023-12-16 19:11:23 -08:00