warn instead

justinchuby · justinchuby · commit 63919ed8a211 · 2023-08-24T04:29:08.000Z
diff --git a/torch/onnx/utils.py b/torch/onnx/utils.py
@@ -1526,13 +1526,16 @@ def _export(
 
     # torch.onnx.export does not support opset versions >=18
     if opset_version > _constants.ONNX_TORCHSCRIPT_EXPORTER_MAX_OPSET:
-        raise ValueError(
+        # We do not want to fail because we should still allow users to create
+        # custom symbolic functions for opset>17
+        warnings.warn(
             f"Exporting to ONNX opset version {opset_version} is not supported. "
             f"by 'torch.onnx.export()'. "
             f"The highest opset version supported is {_constants.ONNX_TORCHSCRIPT_EXPORTER_MAX_OPSET}. "
             f"To use a newer opset version, consider 'torch.onnx.dynamo_export()'. "
             f"Note that dynamo_export() is in preview. Please report errors with "
-            f"dynamo_export() as Github issues to https://github.com/pytorch/pytorch/issues."
+            f"dynamo_export() as Github issues to https://github.com/pytorch/pytorch/issues.",
+            category=errors.OnnxExporterWarning,
         )
 
     if export_modules_as_functions and opset_version < 15:
