Tri Dao
|
79bd1a2d5d
[LayerNorm] Implement residual + LayerNorm/RMSNorm in Triton
|
1 vuosi sitten |
Tri Dao
|
f1a73d0740
Run isort and black on python files
|
1 vuosi sitten |
Tri Dao
|
60499abcfd
[Benchmark] Add script to benchmark FlashAttention
|
1 vuosi sitten |
Tri Dao
|
fb88e5e4b3
Move benchmark utils, support AMP
|
2 vuotta sitten |