pytorch
9ab6ac5b - [ONNX] Fix aten::new_zeros due to TorchScript behavior change on Pytorch 2.1 Fix #110935 (#110956)

Commit
1 year ago
[ONNX] Fix aten::new_zeros due to TorchScript behavior change on Pytorch 2.1 Fix #110935 (#110956) Fixes #110597 Summary: * Generic code: The `torch._C.Value.node().mustBeNone()` is encapsulated into the high-level API `JitScalarType.from_value` ; `_is_none` was also extended to allow either `None` or `torch._C.Value.node.mustBeNone()`, so users don't manually call into TorchScript API when implementing operators * Specific to `new_zeros` (and ops of ` *_like` and `new_*`): When checking `dtype`, we always must use ` _is_none`, which will call proposed by #110935 Pull Request resolved: https://github.com/pytorch/pytorch/pull/110956 Approved by: https://github.com/justinchuby, https://github.com/BowenBao
Author
Thiago Crepaldi
Committer
Parents
Loading