-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Add torch.memory_format support to the TorchScript #28544
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Add torch.memory_format support to the TorchScript #28544
Conversation
[ghstack-poisoned]
…"Add torch.memory_format support to the TorchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
… TorchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…mat support to the TorchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
zdevito
left a comment
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
It's hard to know if this is correct without tests, but the code looks simple enough.
|
Yeah a test, or even just some basic indication how you figured out this was working (e.g., does a later PR exercise it?) would be great. |
…orchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…hScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…the TorchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…ript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
|
Test added |
…he TorchScript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…cript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…cript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
…cript" Differential Revision: [D18093801](https://our.internmc.facebook.com/intern/diff/D18093801) [ghstack-poisoned]
|
@VitalyFedyunin merged this pull request in 5f51037. |
Stack from ghstack:
resize_op. #28292 Add memory format support to theresize_op.operator==of TensorOptions as confusing one #28076 Killoperator==of TensorOptions as confusing oneresize_as_operator #27979 Add memory format support toresize_as_operatorDifferential Revision: D18093801