a d@sdZddlmZddlZddlmZddlmZddlm Z ddl m Z gdZ Gd d d e ZGd d d eZGd ddeZGdddeZGdddeZGdddeZdS)zONNX exporter exceptions.) annotationsN)Optional)_C) _constants) diagnostics)OnnxExporterErrorOnnxExporterWarningCallHintViolationWarning CheckerErrorUnsupportedOperatorErrorSymbolicValueErrorc@seZdZdZdS)rz1Base class for all warnings in the ONNX exporter.N__name__ __module__ __qualname____doc__rrZ/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/torch/onnx/errors.pyrsrc@seZdZdZdS)r zCWarning raised when a type hint is violated during a function call.Nr rrrrr sr c@seZdZdZdS)rz#Errors raised by the ONNX exporter.Nr rrrrr!src@seZdZdZdS)r z2Raised when ONNX checker detects an invalid model.Nr rrrrr 'sr cs*eZdZdZddddfdd ZZS)r z7Raised when an operator is unsupported by the exporter.strintz Optional[int])nameversionsupported_versioncs|dur2tjj}||||}t|tjj|nn|dsP|dsP|dr|tjj}|||t j }t|tjj|n$tjj }||}t|tjj|t |dS)Nzaten::zprim::z quantized::)rrulesZ)operator_supported_in_newer_opset_versionZformat_messageZdiagnoselevelsERROR startswithZ"missing_standard_symbolic_functionrZPYTORCH_GITHUB_ISSUES_URLZ missing_custom_symbolic_functionsuper__init__)selfrrrZdiagnostic_rulemsg __class__rrr0s( z!UnsupportedOperatorError.__init__rrrrr __classcell__rrr!rr -sr cs(eZdZdZdddfdd ZZS)r z+Errors around TorchScript values and nodes.rz_C.Value)r valuec s|d|d|d|d}|}|rH|d|d7}zl|d7}|tdd d d t|Dp|d d d d dd t| Dpd d7}Wnt y|d7}Yn0t |dS)Nz [Caused by the value 'z ' (type 'z;') in the TorchScript graph. The containing node has kind 'z'.] z (node defined in )z zInputs:  css.|]&\}}d|d|d|dVqdSz #z: z (type 'z')Ntype).0iZinput_rrr ]sz.SymbolicValueError.__init__..z Emptyz Outputs: css.|]&\}}d|d|d|dVqdSr(r))r+r,outputrrrr-fsz zv Failed to obtain its input and output for debugging. Please refer to the TorchScript graph for debugging information.) r*nodekindZ sourceRangetextwrapindentjoin enumerateinputsoutputsAttributeErrorrr)rr r%messageZ code_locationr!rrrLs@        zSymbolicValueError.__init__r#rrr!rr Isr )r __future__rr1typingrZtorchrZ torch.onnxrZtorch.onnx._internalr__all__ UserWarningrr RuntimeErrorrr r r rrrrs