pytorch
pytorch copied to clipboard
Move vmap's OrderedDict pytree support to torch.utils._pytree
Stack from ghstack:
- #83079
- -> #83073
There's no reason why it should just apply when you import vmap
Test Plan:
- added a new test
:link: Helpful links
- :test_tube: See artifacts and rendered test results at hud.pytorch.org/pr/83073
- :page_facing_up: Preview Python docs built from this PR
- :page_facing_up: Preview C++ docs built from this PR
- :question:Need help or want to give feedback on the CI? Visit our office hours
:white_check_mark: No Failures (0 Pending)
As of commit 059f1542f6 (more details on the Dr. CI page):
Expand to see more
:green_heart: :green_heart: Looks good so far! There are no failures yet. :green_heart: :green_heart:
This comment was automatically generated by Dr. CI (expand for details).
Please report bugs/suggestions to the (internal) Dr. CI Users group.