Skip to content

Support full graph compilation of foreach Adamax #117812

New issue
New issue
Closed
Closed
Support full graph compilation of foreach Adamax#117812
Assignees
mlazos
Labels
module: dynamomodule: optimizerRelated to torch.optimmodule: pt2 optimizerRelating to torch.compile'd optimoncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
@mlazos

Description

@mlazos
mlazos
opened on Jan 19, 2024

🚀 The feature, motivation and pitch

Refactor Adamax to support multitensor full graph compilation with torch.compile.

WIP PR #110345

I'm going to take this over and land it

Alternatives

No response

Additional context

No response

cc @vincentqb @jbschlosser @albanD @janeyx99 @crcrpar @ezyang @chauhang @penguinwu @voznesenskym @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @amjames @aakhundov

Metadata

Metadata

Assignees

Labels

module: dynamomodule: optimizerRelated to torch.optimmodule: pt2 optimizerRelating to torch.compile'd optimoncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate module

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions