o Æ µiï!ã@s´ddlZddlmZddlZddlZddlmZe d¡e  d¡Z Gdd„dej ƒZ e ƒZ Gdd „d ƒZGd d „d ƒZGd d „d ƒZGdd„dƒZGdd„dƒZGdd„dƒZdS)éN)ÚAny)ÚENABLED_FEATURESTz,torch_tensorrt [TensorRT Conversion Context]c@s0eZdZd dd„Zdejjdeddfdd„ZdS) Ú _TRTLoggerÚreturnNcCstj |¡dS©N)ÚtrtÚILoggerÚ__init__)Úself©r úJ/home/ubuntu/.local/lib/python3.10/site-packages/torch_tensorrt/logging.pyr sz_TRTLogger.__init__ÚseverityÚmsgcCs”|tjjjkrt |¡t|ƒ‚|tjjjkrt |¡dS|tjjj kr,t  |¡dS|tjjj kr:t  |¡dS|tjjj krHt |¡dSdSr)rrÚSeverityÚINTERNAL_ERRORÚ_LOGGERÚcriticalÚ RuntimeErrorÚERRORÚerrorÚWARNINGÚwarningÚINFOÚinfoÚVERBOSEÚdebug)r r rr r r Úlogs ÿz_TRTLogger.log©rN) Ú__name__Ú __module__Ú __qualname__r rrrÚstrrr r r r r s rc@ó4eZdZdZd dd„Zdedededdfd d „ZdS) Úinternal_errorszãContext-manager to limit displayed log messages to just internal errors Example: .. code-block:: py with torch_tensorrt.logging.internal_errors(): outputs = model_torchtrt(inputs) rNcCóxt ¡|_t tj¡tjr"ddlm}|  ¡|_ |  |j j ¡dStjr:tjj ¡|_tjj ttjjjƒ¡dSdS©Nr©Úlogging)rÚgetEffectiveLevelÚ external_lvlÚsetLevelr'ÚCRITICALrÚtorchscript_frontendÚtorch_tensorrt.tsÚget_reportable_log_levelÚts_levelÚset_reportable_log_levelÚLevelÚ InternalErrorÚtorch_tensorrt_runtimeÚtorchÚopsÚtensorrtÚget_logging_levelÚrt_levelÚset_logging_levelÚintrrrr©r Ú ts_loggingr r r Ú __enter__-s     ÿþzinternal_errors.__enter__Úexc_typeÚ exc_valueÚexc_tbcCóLt |j¡tjrddlm}| |j¡dStj r$t j j   |j¡dSdSr%©rr*r)rr,r-r'r0r/r3r4r5r6r9r8©r r>r?r@r<r r r Ú__exit__=ó  ÿzinternal_errors.__exit__r©rrr Ú__doc__r=rrDr r r r r#"s r#c@r") ÚerrorszÛContext-manager to limit displayed log messages to just errors and above Example: .. code-block:: py with torch_tensorrt.logging.errors(): outputs = model_torchtrt(inputs) rNcCóxt ¡|_t tj¡tjr"ddlm}|  ¡|_ |  |j j ¡dStjr:tjj ¡|_tjj ttjjjƒ¡dSdSr%)rr(r)r*r'rrr,r-r.r/r0r1ÚErrorr3r4r5r6r7r8r9r:rrrr;r r r r=Tó    þzerrors.__enter__r>r?r@cCrAr%rBrCr r r rDbrEzerrors.__exit__rrFr r r r rHIó rHc@r") ÚwarningszðContext-manager to limit displayed log messages to just warnings and above Example: .. code-block:: py with torch_tensorrt.logging.warnings(): model_trt = torch_tensorrt.compile(model, **spec) rNcCrIr%)rr(r)r*r'rrr,r-r.r/r0r1ÚWarningr3r4r5r6r7r8r9r:rrrr;r r r r=yrKzwarnings.__enter__r>r?r@cCrAr%rBrCr r r rD‡rEzwarnings.__exit__rrFr r r r rMnrLrMc@r") rzãContext-manager to display all info and greater severity messages Example: .. code-block:: py with torch_tensorrt.logging.info(): model_trt = torch_tensorrt.compile(model, **spec) rNcCrIr%)rr(r)r*r'rrr,r-r.r/r0r1ÚInfor3r4r5r6r7r8r9r:rrrr;r r r r=žrKzinfo.__enter__r>r?r@cCrAr%rBrCr r r rD¬rEz info.__exit__rrFr r r r r“rLrc@r") rzçContext-manager to display full debug information through the logger Example: .. code-block:: py with torch_tensorrt.logging.debug(): model_trt = torch_tensorrt.compile(model, **spec) rNcCr$r%)rr(r)r*r'ÚDEBUGrr,r-r.r/r0r1ÚDebugr3r4r5r6r7r8r9r:rrrrr;r r r r=ÃrKzdebug.__enter__r>r?r@cCrAr%rBrCr r r rDÑrEzdebug.__exit__rrFr r r r r¸rLrc@r") Úgraphsa#Context-manager to display the results of intermediate lowering passes as well as full debug information through the logger Example: .. code-block:: py with torch_tensorrt.logging.graphs(): model_trt = torch_tensorrt.compile(model, **spec) rNcCs|t ¡|_t tj¡tjr"ddlm}|  ¡|_ |  |j j ¡dStjrr?r@cCrAr%rBrCr r r rD÷rEzgraphs.__exit__rrFr r r r rRÝs rR)r'Útypingrr6rr4Útorch_tensorrt._featuresrÚcaptureWarningsÚ getLoggerrrrÚ TRT_LOGGERr#rHrMrrrRr r r r Ús    '%%%%