mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
a8574a9719
Prior to this `lintrunner --all-files --take FLAKE8` failed. Pull Request resolved: https://github.com/pytorch/pytorch/pull/124771 Approved by: https://github.com/Skylion007 ghstack dependencies: #124428
57 lines
1.7 KiB
Python
57 lines
1.7 KiB
Python
from enum import Enum
|
|
|
|
|
|
class ExportErrorType(Enum):
|
|
# User providing invalid inputs to either tracer, or other public facing APIs
|
|
INVALID_INPUT_TYPE = 1
|
|
|
|
# User returning values from their models that we don't support.
|
|
INVALID_OUTPUT_TYPE = 2
|
|
|
|
# Generated IR does not conform to Export IR Specification.
|
|
VIOLATION_OF_SPEC = 3
|
|
|
|
# User's code contains types and functionalities we don't support.
|
|
NOT_SUPPORTED = 4
|
|
|
|
# User's code didn't provide necessary details for us to successfully trace and export.
|
|
# For example, we use a lot of decorators and ask users to annotate their model.
|
|
MISSING_PROPERTY = 5
|
|
|
|
# User is using an API without proper initialization step.
|
|
UNINITIALIZED = 6
|
|
|
|
|
|
def internal_assert(pred: bool, assert_msg: str) -> None:
|
|
"""
|
|
This is exir's custom assert method. It internally just throws InternalError.
|
|
Note that the sole purpose is to throw our own error while maintaining similar syntax
|
|
as python assert.
|
|
"""
|
|
|
|
if not pred:
|
|
raise InternalError(assert_msg)
|
|
|
|
|
|
class InternalError(Exception):
|
|
"""
|
|
Raised when an internal invariance is violated in EXIR stack.
|
|
Should hint users to report a bug to dev and expose the original
|
|
error message.
|
|
"""
|
|
|
|
def __init__(self, message: str) -> None:
|
|
super().__init__(message)
|
|
|
|
|
|
class ExportError(Exception):
|
|
"""
|
|
This type of exception is raised for errors that are directly caused by the user
|
|
code. In general, user errors happen during model authoring, tracing, using our public
|
|
facing APIs, and writing graph passes.
|
|
"""
|
|
|
|
def __init__(self, error_code: ExportErrorType, message: str) -> None:
|
|
prefix = f"[{error_code}]: "
|
|
super().__init__(prefix + message)
|