Add tests for aliasing in fake tensor#82337
Add tests for aliasing in fake tensor#82337eellison wants to merge 7 commits intogh/eellison/300/basefrom
Conversation
[ghstack-poisoned]
🔗 Helpful links
✅ No Failures (0 Pending)As of commit d71d0bc (more details on the Dr. CI page): Expand to see more💚 💚 Looks good so far! There are no failures yet. 💚 💚 This comment was automatically generated by Dr. CI (expand for details).Please report bugs/suggestions to the (internal) Dr. CI Users group. |
| if isinstance(out, torch.Tensor): | ||
| input_storages.add(out.storage()._cdata) | ||
| for inp in tree_flatten(inputs)[0]: | ||
| if isinstance(inp, torch.Tensor) and inp.storage()._cdata in input_storages: |
There was a problem hiding this comment.
should this use StorageWeakRef(inp.storage()) instead of a private API 😛 (I guess it doesn't matter much since they do the same thing).
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
[ghstack-poisoned]
|
@pytorchbot merge |
|
@pytorchbot successfully started a merge job. Check the current status here |
|
Hey @eellison. |
Summary: Pull Request resolved: #82337 Approved by: https://github.com/ezyang, https://github.com/bdhirsh Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/b2f6aa666e86ed1369f7bfce4e9adab7ae4b98d4 Reviewed By: kit1980 Differential Revision: D38359256 Pulled By: eellison fbshipit-source-id: 430eacb55ec04a3df16739ad933580b8f10a1a7f
Stack from ghstack (oldest at bottom):