github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-02 01:28:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

[release] update version (#6041)

Browse Source 
* [release] update version

* [devops] update comp test

* [devops] update comp test debug

* [devops] debug comp test

* [devops] debug comp test

* [devops] debug comp test

* [devops] debug comp test

* [devops] debug comp test
This commit is contained in:
Hongxin Liu
2024-09-10 10:31:09 +08:00
committed by GitHub
parent 5ce6dd75bf
commit b3db1058ec
10 changed files with 10 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_infer/test_kernels/triton/test_fused_rotary_embedding.py
View File

@@ -19,6 +19,7 @@ except ImportError:
TRITON_CUDA_SUPPORT = version.parse(torch.version.cuda) > version.parse("11.4")
@pytest.mark.skip(reason="cuda error")
@pytest.mark.skipif(not (HAS_TRITON and TRITON_CUDA_SUPPORT), reason="requires triton")
def test_fused_rotary_emb():
num_tokens = 20