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
|
93383bd55b
[TP] Implement TensorParallel without sequence parallel
|
1 year ago |
Tri Dao
|
b4018a5028
Implement Tensor Parallel for GPT model
|
2 years ago |
Tri Dao
|
a8cfe51551
Implement Tensor Parallel for transformer Block
|
2 years ago |