pytorch

d00de0d4 - Support dataclasses in TorchScript (#73066)

Support dataclasses in TorchScript (#73066) Summary: Fixes https://github.com/pytorch/pytorch/issues/72901. Since we can't get access to the source code for synthesized magic methods on dataclasses, we have to synthesize our own versions. `torch/jit/_dataclass_impls.py` has the code that does this. What's supported - Synthesized `__init__`, `__eq__`, and the comparison magic methods when `order=True` is set on the dataclass decorator - Default values for fields - `__post_init__`, including using `InitVar` fields inside of `__post_init__`, on Python 3.8+ - Overriding `__eq__` or any of the comparison magic methods to provide your own implementation What's not supported - Default factory initializers for fields - Frozen dataclasses - `InitVar` on Python 3.7 - `__repr__` and `__hash__` (these are actually implemented, but the TorchScript interpreter won't call them) - Using the `!=` operator on dataclasses inside TorchScript; this is because TorchScript requires that you implement `__ne__` to use this operator, whereas in regular Python the `!=` operator will resolve to the negation of whatever is returned by `__eq__` if there's no `__ne__`. Dataclasses don't actually synthesize an `__ne__` method for this reason. I've been toying with different ways to fix this but `!=` is not working in this PR at the moment. qihqi Pull Request resolved: https://github.com/pytorch/pytorch/pull/73066 Reviewed By: mrshenli Differential Revision: D34398107 Pulled By: qihqi fbshipit-source-id: f5a792555c88f3631f97837a96687e4890660a32 (cherry picked from commit ea7f077dc49a4ee75ca0d1409aedd85228952881)