pytorch
ff7d5b62 - Improve ProxyTensor tensor_tree list/tuple handling (#99897)

Commit
1 year ago
Improve ProxyTensor tensor_tree list/tuple handling (#99897) This PR improves the list/tuple handling by merging the logic into `wrap_with_proxy` directly, and set_meta when we find the current proxy is a fx.Proxy. This also solves the problem that even `fused_adam` have `val`, some corresponding `getitem` calls followed after `fused_adam` don't have val Pull Request resolved: https://github.com/pytorch/pytorch/pull/99897 Approved by: https://github.com/ezyang
Author
Committer
Parents
Loading