ColossalAI/tests/test_infer/test_models
Jianghai e545a871b8 [Hotfix] Fix accuracy and align attention method api with Triton kernel (#5229)
* fix accuracy

* alignment in attention

* fix attention

* fix

* fix bugs

* fix bugs

* fix bugs
2024-01-11 13:46:14 +00:00
..
test_attention.py [Hotfix] Fix accuracy and align attention method api with Triton kernel (#5229) 2024-01-11 13:46:14 +00:00