Skip to content

Commit

Permalink
Implement a post_compile step for aot_dispatch_autograd (#126193)
Browse files Browse the repository at this point in the history 
This PR moves the post compile portion of aot_dispatch_autograd into runtime_wrappers.py. Completing this allows us to run the post compile section on its own when warm starting.

I considered leaving this thing in jit_compile_runtime_wrappers, but we're gonna run into circular dependency issues later if we don't move it over
Pull Request resolved: #126193
Approved by: https://github.com/bdhirsh
ghstack dependencies: #126907
  • Loading branch information
@jamesjwu @pytorchmergebot
jamesjwu authored and pytorchmergebot committed May 25, 2024
1 parent 52bcf12 commit cc9a341
Show file tree
Hide file tree
Showing 2 changed files with 661 additions and 585 deletions.
Loading

0 comments on commit cc9a341

Please sign in to comment.