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

change torch._dynamo.export(aten_graph=...) to allow pre_autograd tracing #98031

Closed
wants to merge 12 commits into from

Commits on Mar 30, 2023

  1. change torch._dynamo.export(aten_graph=...) to allow pre_autograd tra…

    …cing
    
    [ghstack-poisoned]
    bdhirsh committed Mar 30, 2023
    Configuration menu
    Copy the full SHA
    d7e356b View commit details
    Browse the repository at this point in the history

Commits on Apr 11, 2023

  1. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 11, 2023
    Configuration menu
    Copy the full SHA
    9d694fa View commit details
    Browse the repository at this point in the history

Commits on Apr 12, 2023

  1. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 12, 2023
    Configuration menu
    Copy the full SHA
    b2de3a4 View commit details
    Browse the repository at this point in the history
  2. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 12, 2023
    Configuration menu
    Copy the full SHA
    423561f View commit details
    Browse the repository at this point in the history
  3. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 12, 2023
    Configuration menu
    Copy the full SHA
    b489623 View commit details
    Browse the repository at this point in the history

Commits on Apr 13, 2023

  1. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    b5ec040 View commit details
    Browse the repository at this point in the history
  2. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    b51acfb View commit details
    Browse the repository at this point in the history

Commits on Apr 24, 2023

  1. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    61c21be View commit details
    Browse the repository at this point in the history
  2. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    74d8e07 View commit details
    Browse the repository at this point in the history

Commits on Apr 25, 2023

  1. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    172d7c1 View commit details
    Browse the repository at this point in the history
  2. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    4653b55 View commit details
    Browse the repository at this point in the history
  3. Update on "change torch._dynamo.export(aten_graph=...) to allow pre_a…

    …utograd tracing"
    
    pre_autograd tracing is still early, but it should work for basic cases. This PR changes the API a bit for export to expose pre_autograd tracing. Name bikeshedding is welcome, but it looks like:
    ```
    torch._dynamo.export(..., aten_graph="aten_pre_autograd")
    ```
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire andrewor14 
    
    
    
    
    cc soumith voznesenskym penguinwu anijain2305 EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng Xia-Weiwen wenzhe-nrv jiayisunx desertfire
    
    [ghstack-poisoned]
    bdhirsh committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    7f6bdb5 View commit details
    Browse the repository at this point in the history