Tri Dao
|
79bd1a2d5d
[LayerNorm] Implement residual + LayerNorm/RMSNorm in Triton
|
hai 1 ano |
Tri Dao
|
f1a73d0740
Run isort and black on python files
|
hai 1 ano |
Tri Dao
|
60499abcfd
[Benchmark] Add script to benchmark FlashAttention
|
hai 1 ano |
Tri Dao
|
fb88e5e4b3
Move benchmark utils, support AMP
|
%!s(int64=2) %!d(string=hai) anos |