Skip to content
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

Allow Lazy Initialization for ManualPipelineStage #126216

Open
H-Huang opened this issue May 14, 2024 · 0 comments
Open

Allow Lazy Initialization for ManualPipelineStage #126216

H-Huang opened this issue May 14, 2024 · 0 comments
Assignees
@H-Huang
Labels
oncall: distributed Add this issue/PR to distributed oncall triage queue triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Comments

@H-Huang
Copy link
Member

H-Huang commented May 14, 2024 •
edited by pytorch-bot bot

Using this issue to track the requests in #126123

Some requests:

  • manual pipeline stage, shape inference on first forward.
  • Remove device requirement.
  • Do not require output_args

cc @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @osalpekar @jiayisuse @kwen2501 @awgu @penguinwu @fegin @XilunWu @wanchaol @fduwjj @wz337 @tianyu-l @wconstab @yf225 @chauhang @d4l3k
@H-Huang H-Huang added oncall: distributed Add this issue/PR to distributed oncall triage queue triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module labels May 14, 2024
@H-Huang H-Huang self-assigned this May 14, 2024
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Assignees

@H-Huang H-Huang

Labels
oncall: distributed Add this issue/PR to distributed oncall triage queue triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module
Projects
None yet
Milestone
No milestone
Development

No branches or pull requests
1 participant
@H-Huang