Skip to content

GitLab

Unverified Commit 51597f6a authored by Jiarui Fang's avatar Jiarui Fang Committed by GitHub
Browse files

[hotfix] pass test_complete_workflow (#1877)

parent 986f8cba
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tests/test_fx/test_complete_workflow.py
View file @ 51597f6a
...@@ -50,7 +50,7 @@ def run_workflow(world_size, dev): ...@@ -50,7 +50,7 @@ def run_workflow(world_size, dev):
annotated_gm.recompile() annotated_gm.recompile()
# materialization and sharding # materialization and sharding
ctx.lazy_init_parameters(annotated_gm) ctx.lazy_init_parameters(annotated_gm, device=dev)
for param in model.parameters(): for param in model.parameters():
assert not param.is_meta assert not param.is_meta
...@@ -84,4 +84,4 @@ def test_complete_workflow(world_size, dev): ...@@ -84,4 +84,4 @@ def test_complete_workflow(world_size, dev):
if __name__ == '__main__': if __name__ == '__main__':
test_complete_workflow(1) test_complete_workflow(1, 'cuda')
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment