New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

[autoparallel] adapt solver and CostGraph with new handler #1695

Merged

FrankLeeeee merged 3 commits into hpcaitech:main from YuliangLiu0306:feature/solver_v2

Oct 13, 2022

Contributor

YuliangLiu0306 commented Oct 13, 2022

What does this PR do

In latest handler, all costs are in type of TrainCycleItem, so Solver and CostGraph are changed to work with the latest handler.
Assign some torch operations to handlers.
Use Solver_V2 and CostGraph_V2 on resnet50 model, and get wanted seaching result.


          [autoparallel] adapt solver and CostGraph with new handler

d268ba1

FrankLeeeee added the Run Build and Test label

YuliangLiu0306 and others added 2 commits

October 13, 2022 13:43


          Merge branch 'main' into feature/solver_v2

75ca398


          fix test issue

06d279a

FrankLeeeee approved these changes

View reviewed changes

FrankLeeeee merged commit 81f7530 into hpcaitech:main

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment