github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

update codegen test

pull/2364/head
oahzxl 2023-01-09 14:57:47 +08:00
parent 3abbaf8bc6
commit a005965d2d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tests/test_autochunk/test_autochunk_codegen.py
View File

@ -99,7 +99,7 @@ def _test_autochunk_codegen(rank, msa_len, pair_len, max_memory):
gpc.destroy() gpc.destroy()
@pytest.mark.parametrize("max_memory", [None, 20, 24, 28, 32]) @pytest.mark.parametrize("max_memory", [None, 20, 25, 30])
@pytest.mark.parametrize("msa_len", [32]) @pytest.mark.parametrize("msa_len", [32])
@pytest.mark.parametrize("pair_len", [64]) @pytest.mark.parametrize("pair_len", [64])
def test_autochunk_codegen(msa_len, pair_len, max_memory): def test_autochunk_codegen(msa_len, pair_len, max_memory):