squall/flash-attention
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity
665b55e2e2
flash-attention/tests/ops/triton
History
Tri Dao 665b55e2e2 [LayerNorm] Implement parallel layer norm in Triton
2024-01-04 23:15:35 -08:00
..
test_layer_norm.py [LayerNorm] Implement parallel layer norm in Triton 2024-01-04 23:15:35 -08:00