Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
a005965d
Commit
a005965d
authored
Jan 09, 2023
by
oahzxl
Browse files
update codegen test
parent
3abbaf8b
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
tests/test_autochunk/test_autochunk_codegen.py
tests/test_autochunk/test_autochunk_codegen.py
+1
-1
No files found.
tests/test_autochunk/test_autochunk_codegen.py
View file @
a005965d
...
...
@@ -99,7 +99,7 @@ def _test_autochunk_codegen(rank, msa_len, pair_len, max_memory):
gpc
.
destroy
()
@
pytest
.
mark
.
parametrize
(
"max_memory"
,
[
None
,
20
,
2
4
,
28
,
3
2
])
@
pytest
.
mark
.
parametrize
(
"max_memory"
,
[
None
,
20
,
2
5
,
3
0
])
@
pytest
.
mark
.
parametrize
(
"msa_len"
,
[
32
])
@
pytest
.
mark
.
parametrize
(
"pair_len"
,
[
64
])
def
test_autochunk_codegen
(
msa_len
,
pair_len
,
max_memory
):
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment