Skip to content

GitLab

".github/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "05b8a8de58da5f6a7853b17488cc5b7713d06f38"
Unverified Commit 9eae8554 authored by YuliangLiu0306's avatar YuliangLiu0306 Committed by GitHub
Browse files

[hotfix] add recompile after graph manipulatation (#1621)

parent d967779a
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 0 deletions
tests/test_auto_parallel/test_shape_consistency_pass.py
View file @ 9eae8554
...@@ -65,6 +65,7 @@ def check_apply(rank, world_size, port): ...@@ -65,6 +65,7 @@ def check_apply(rank, world_size, port):
solution = list(ret[0]) solution = list(ret[0])
sharding_spec_dict, origin_spec_dict = solution_annotatation_pass(gm, solution, device_mesh) sharding_spec_dict, origin_spec_dict = solution_annotatation_pass(gm, solution, device_mesh)
shape_consistency_pass(gm) shape_consistency_pass(gm)
gm.recompile()
nodes = [node for node in gm.graph.nodes] nodes = [node for node in gm.graph.nodes]
# TODO: wrap the gm to avoid the influence of the user training code # TODO: wrap the gm to avoid the influence of the user training code
output = gm(input, sharding_spec_dict, origin_spec_dict) output = gm(input, sharding_spec_dict, origin_spec_dict)
......
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