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

[hotfix] pass test_complete_workflow (#1877)

Browse Source
This commit is contained in:
Jiarui Fang
2022-11-10 17:53:39 +08:00
committed by GitHub
parent 986f8cbaa7
commit 51597f6a28
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_fx/test_complete_workflow.py
View File

@@ -50,7 +50,7 @@ def run_workflow(world_size, dev):
annotated_gm.recompile()
# materialization and sharding
ctx.lazy_init_parameters(annotated_gm)
ctx.lazy_init_parameters(annotated_gm, device=dev)
for param in model.parameters():
assert not param.is_meta
@@ -84,4 +84,4 @@ def test_complete_workflow(world_size, dev):
if __name__ == '__main__':
test_complete_workflow(1)
test_complete_workflow(1, 'cuda')