mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
[dynamo] Guard serialization for FUNCTORCH_STACK_MATCH (#152616)
Make Functorch interpreters serializable most of the time, so that we can save the guards on functorch states. ## Test Cases: 0. torch.compile() without functorch layers present. Guard should fail with any layer being pushed. 1. torch.compile() nested in vmap. 2. torch.compile() nested in grad. 3. torch.compile() nested in jvp + vmap 4. torch.compile() nested functionalize 5. torch.compile() nested in vmap + grad Differential Revision: [D74008787](https://our.internmc.facebook.com/intern/diff/D74008787/) Pull Request resolved: https://github.com/pytorch/pytorch/pull/152616 Approved by: https://github.com/zou3519 ghstack dependencies: #152615
This commit is contained in:
committed by
PyTorch MergeBot
parent
1d1cbcd8a3
commit
ffd58293f7
@ -376,6 +376,7 @@ cc_library(
|
||||
":torch_headers",
|
||||
"@fbgemm",
|
||||
"@ideep",
|
||||
"@nlohmann",
|
||||
],
|
||||
alwayslink = True,
|
||||
)
|
||||
|
Reference in New Issue
Block a user