[FSDP][BE] Improve the assert message of sharded load_state_dict#87486
[FSDP][BE] Improve the assert message of sharded load_state_dict#87486fegin wants to merge 5 commits intogh/fegin/35/basefrom
Conversation
[ghstack-poisoned]
🔗 Helpful Links🧪 See artifacts and rendered test results at hud.pytorch.org/pr/87486
Note: Links to docs will display an error until the docs builds have been completed. ❌ 1 Failures, 6 PendingAs of commit ab8af39: This comment was automatically generated by Dr. CI and updates every 15 minutes. |
| ) | ||
|
|
||
| from ._fsdp_extensions import _ext_chunk_tensor, _ext_pre_load_state_dict_transform | ||
| from ._fsdp_extensions import _extensions as _users_extensions |
There was a problem hiding this comment.
nit: Maybe better?
| from ._fsdp_extensions import _extensions as _users_extensions | |
| from ._fsdp_extensions import _extensions as _user_extensions |
…e_dict" [ghstack-poisoned]
…e_dict" [ghstack-poisoned]
…e_dict" [ghstack-poisoned]
…e_dict" [ghstack-poisoned]
|
@pytorchbot merge -f "The failing test is not related". |
Merge startedYour change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Learn more about merging in the wiki. Questions? Feedback? Please reach out to the PyTorch DevX Team |
|
Hey @fegin. |
…orch#87486) Pull Request resolved: pytorch#87486 Approved by: https://github.com/awgu
…orch#87486) Pull Request resolved: pytorch#87486 Approved by: https://github.com/awgu
Stack from ghstack (oldest at bottom):