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.

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

AOT Backend #237

Open
shivadbhavsar opened this issue Mar 11, 2025 · 0 comments · May be fixed by #235
Open

AOT Backend #237

shivadbhavsar opened this issue Mar 11, 2025 · 0 comments · May be fixed by #235
Assignees
Labels
TorchMIGraphX For internal tracking

Comments

@shivadbhavsar
Copy link
Collaborator

Currently the migraphx torch.compile backend relies on aot_export_joint_simple to generate an aten graph that can be interpreted to migraphx IR. There are some limitations to this API, particularly for handling graphs that mutate input tensors.
The AotAutograd plugin (used by Inductor) is better built to handle these cases. We should create a migraphx backend that uses aot autograd for generating the aten IR. Initially, this can be an optional/experimental backend to only be used with particular models that require this functionallity.

@shivadbhavsar shivadbhavsar self-assigned this Mar 11, 2025
@shivadbhavsar shivadbhavsar added the TorchMIGraphX For internal tracking label Mar 11, 2025
@shivadbhavsar shivadbhavsar linked a pull request Mar 11, 2025 that will close this issue
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
TorchMIGraphX For internal tracking
Projects
None yet
Development

Successfully merging a pull request may close this issue.

1 participant