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

[hotfix] fix autoparallel compatibility test issues (#2754)

Browse Source
This commit is contained in:
YuliangLiu0306
2023-02-23 17:28:36 +08:00
committed by GitHub
parent 0f392d7403
commit 819e25d8b1
3 changed files with 23 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/auto_parallel/tensor_shard/initialize.py
View File

@@ -330,6 +330,7 @@ def autoparallelize(model: nn.Module,
device_mesh,
solver_preference=solver_preference,
dataloader_option=dataloader_option,
shard_option=shard_option,
save_solver_solution=save_solver_solution,
load_solver_solution=load_solver_solution,
solution_path=solver_solution_path,