pytorch
690bc1c5 - [ONNX] Raise exception for unimplemented ops for non-caffe2 builds

Commit
2 years ago
[ONNX] Raise exception for unimplemented ops for non-caffe2 builds Currently, when an operator symbolic hits an unimplemented scenario, the symbolic may print a warning and return, allowing a non-ONNX operator be emitted into the graph This PRs maintains this behavior for 1) Caffe2 builds or 2) non-caffe2 builds with `operator_export_type != ONNX`. If none of the conditions above are met, the converter raises a `RuntimeError` exception otherwise. This is needed so that exporter can detect detect unsupported ONNX operators when ATEN fallback is used (for non-caffe2 scenarios) Pull Request resolved: https://github.com/pytorch/pytorch/pull/75468 Approved by: https://github.com/BowenBao
Author
Thiago Crepaldi
Committer
Parents
Loading