.. |
trsrc
|
e5bbc2e559
Sparse attn + ops/runtime refactor + v0.3.0 (#343)
|
4 years ago |
__init__.py
|
6996bb0159
Sparse attn triton v1.0 support + torch1.8 test runner (#1374)
|
3 years ago |
bert_sparse_self_attention.py
|
be789b1665
Fix many typos (#1423)
|
3 years ago |
matmul.py
|
3ed77304df
Fix sparse attention for small block-sizes (#1545)
|
2 years ago |
softmax.py
|
6996bb0159
Sparse attn triton v1.0 support + torch1.8 test runner (#1374)
|
3 years ago |
sparse_attention_utils.py
|
be789b1665
Fix many typos (#1423)
|
3 years ago |
sparse_self_attention.py
|
80e263c571
fix sparse_attention imports (#1446)
|
3 years ago |
sparsity_config.py
|
be789b1665
Fix many typos (#1423)
|
3 years ago |