mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
751c54b546
PT2 Export API Prototype Pull Request resolved: https://github.com/pytorch/pytorch/pull/100034 Approved by: https://github.com/angelayi
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)
|