o Xi@sddlmZddgZddlZddlZddlmmZddl m Z ddl m Z m Z eeZGddde jjZGdd d e jjZ ddddZdS)) annotationsConvertVersionPassconvert_versionN)ir) _c_api_utils_version_convertercs.eZdZdZddfd d ZdddZZS)raConvert the model to the specified ONNX opset version. This pass leverages the onnxscript version converter to convert the model. If the conversion is not supported, it falls back to the onnx C API to convert the model. This pass is in-place. The pass is an no-op if the c-api fails. Attributes: target_version: The target ONNX opset version to convert the model to. fallback: Whether to fallback to the onnx version converter if the target version is not supported. Default is False. Ftarget_versionintfallbackboolreturnNonecsBt||_||_tjt||dt t t |_ dS)Nrr ) super__init__rr rpasses Sequential_ConvertVersionPass common_passesRemoveUnusedNodesPassRemoveUnusedFunctionsPassRemoveUnusedOpsetsPass convert_passselfrr  __class__Y/home/ubuntu/.local/lib/python3.10/site-packages/onnxscript/version_converter/__init__.pyr$s  zConvertVersionPass.__init__modelir.Modelir.passes.PassResultcCs ||SN)r)rrrrrcall2s zConvertVersionPass.call)Frr r r r r rr r r!__name__ __module__ __qualname____doc__rr# __classcell__rrrrrscs,eZdZdZdfdd Zdd dZZS)ra Convert the model to the specified ONNX opset version. This pass leverages the onnxscript version converter to convert the model. If the conversion is not supported, it falls back to the onnx C API to convert the model. This pass is in-place. The pass is an no-op if the c-api fails. Attributes: target_version: The target ONNX opset version to convert the model to. fallback: Whether to fallback to the onnx version converter if the target version is not supported. rr r r r r cst||_||_dSr")rrrr rrrrrEs  z_ConvertVersionPass.__init__rr r!c sd|jjvr|jjd}|jkrtj|dSjr"t|jr1tj |jdtj|dSjsBt djtj|dSt djdfd d }z t j ||d }Wn"ty{}zt j d j|dtj|dWYd}~Sd}~wwt|}|jjD]}|j|jjvr|jj|jj|_|j|q|jjdt|jj}|jj|jj||j|_tj|dS)NFrTzThe model version conversion is not supported by the onnxscript version converter and fallback is disabled. The model was not modified (target version: %d). Set fallback=True to enable fallback to the onnx c-api version converter.zThe model version conversion is not supported by the onnxscript version converter and fallback is enabled. The model will be converted using the onnx C API (target version: %d).protoonnx.ModelProtor cstjj|jdS)z$Partial function to check the model.r-)onnxversion_converterrr)r.rrr_partial_convert_versionqsz:_ConvertVersionPass.call.._partial_convert_version)funcrzeFailed to convert the model to the target version %d using the ONNX C API. The model was not modified)exc_info)r.r/r r/)graph opset_importsrrr PassResultr rversion_supportedrloggerwarningr call_onnx_api Exception from_protoinputsname initializers const_valueregister_initializerlenclearextend) rronnx_opset_versionr3converted_protoeconverted_modelinput user_inputsrr2rr#Js^         z_ConvertVersionPass.callr$r%r&rrrrr6srrir.Model | onnx.ModelProtorr r r cCszt|tjr|}t|}nd}t|tjsJt||d||dur;|j|j dd=|j t |jdSdS)a,Convert the model to the specified ONNX opset version. Args: model: The model to convert. target_version: The target ONNX opset version. fallback: Whether to fallback to the onnx version converter if the target version is not supported. Default is False. Nr) isinstancer0 ModelProtorr>Modelrr6Clear functionsCopyFromto_proto)rrr model_protorrrrs    r")rrMrr r r ) __future__r__all__loggingr0onnx_ir.passes.commonrcommonr onnxscriptronnxscript.version_converterrr getLoggerr'r: InPlacePassrrrrrrrs   !_