Tri Dao
|
0e8c46ae08
Run isort and black on test files
|
1 year ago |
Tri Dao
|
88173a1aaf
[FusedDense] Support relu, rename FusedDenseGeluDense -> FusedMLP
|
1 year ago |
Tri Dao
|
e68ebbe89a
Simplify FusedDense
|
2 years ago |
Tri Dao
|
fa6d1ce44f
Add fused_dense and dropout_add_layernorm CUDA extensions
|
2 years ago |