Skip to content

Expand AOTI tutorial to cover AMP #3152

Open
@bhack

Description

@bhack

📚 The doc issue

I think it would useful to expand AOTI tutorial to cover AMP trained models.

https://pytorch.org/tutorials/recipes/torch_export_aoti_python.html

Suggest a potential alternative/fix

No response

cc @svekars @brycebortree @sekyondaMeta @ezyang @chauhang @penguinwu @avikchaudhuri @gmagogsfm @zhxchen17 @tugsbayasgalan @angelayi @suo @ydwu4 @desertfire @chenyang78

Activity

bhack

bhack commented on Nov 6, 2024

@bhack
Author
agunapal

agunapal commented on Nov 6, 2024

@agunapal
Contributor

@bhack Thanks for the suggestion. Will check

svekars

svekars commented on Nov 8, 2024

@svekars
Contributor

Let's move this to the tutorials repo.

transferred this issue frompytorch/pytorchon Nov 8, 2024
bhack

bhack commented on Jan 31, 2025

@bhack
Author

What is the status of this with 2.6.0 ?

angelayi

angelayi commented on Apr 16, 2025

@angelayi
Contributor

@bhack what sort of model are you looking at? Could you provide a pointer?
I would think that an AMP trained model behaves the same as just any model that you pass to export?

bhack

bhack commented on Apr 16, 2025

@bhack
Author

Nothing specific just a general guide how to export/aoti compile an amp model.
Mainly not only specifically amp trained but also the case when we want to use amp at inference.

E.g. do we need to amp wrap the model before export/compile? Etc..

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

Metadata

Metadata

Assignees

No one assigned

    Labels

    triagedIssues has been triaged

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

      Development

      No branches or pull requests

        Participants

        @bhack@svekars@angelayi@bdhirsh@agunapal

        Issue actions

          Expand AOTI tutorial to cover AMP · Issue #3152 · pytorch/tutorials