Reza Yazdani
|
3ed77304df
Fix sparse attention for small block-sizes (#1545)
|
2 years ago |
Jeff Rasley
|
6996bb0159
Sparse attn triton v1.0 support + torch1.8 test runner (#1374)
|
3 years ago |
Hyunwoong Ko
|
eb97a424f2
Update matmul.py (#1349)
|
3 years ago |
Reza Yazdani
|
26e3841cd4
Change the sparse attention API to be compatible with latest changes of triton (#902)
|
3 years ago |
Jeff Rasley
|
31f46feee2
DeepSpeed JIT op + PyPI support (#496)
|
4 years ago |
Jeff Rasley
|
e5bbc2e559
Sparse attn + ops/runtime refactor + v0.3.0 (#343)
|
4 years ago |