def register_logger_callback():
from polygraphy.logger import G_LOGGER
def set_tf_logging_level(sev):
import os
from polygraphy import mod
tf = mod.lazy_import("tensorflow", version="<2.0")
if not mod.has_mod(tf, with_attr="__version__"):
return
if sev > G_LOGGER.WARNING:
tf_sev = tf.compat.v1.logging.ERROR
tf_logging_level = "3"
elif sev > G_LOGGER.INFO:
tf_sev = tf.compat.v1.logging.WARN
tf_logging_level = "2"
elif sev > G_LOGGER.VERBOSE:
tf_sev = tf.compat.v1.logging.INFO
tf_logging_level = "1"
else:
tf_sev = tf.compat.v1.logging.DEBUG
tf_logging_level = "0"
tf.compat.v1.logging.set_verbosity(tf_sev)
os.environ['TF_CPP_MIN_LOG_LEVEL'] = tf_logging_level
G_LOGGER.register_callback(
set_tf_logging_level
) # Will be registered when this runner is imported.
def run(self, args):
def set_onnx_gs_logging_level(sev):
ONNX_GS_LOGGER = gs.logger.G_LOGGER
if sev >= G_LOGGER.CRITICAL:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.CRITICAL
elif sev >= G_LOGGER.ERROR:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.ERROR
elif sev >= G_LOGGER.WARNING:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.WARNING
elif sev >= G_LOGGER.INFO:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.INFO
elif sev >= G_LOGGER.EXTRA_VERBOSE:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.DEBUG
elif sev >= G_LOGGER.SUPER_VERBOSE:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.VERBOSE
else:
ONNX_GS_LOGGER.severity = ONNX_GS_LOGGER.ULTRA_VERBOSE
fmts = self.arg_groups[LoggerArgs].log_format
for fmt in fmts:
if fmt == "no-colors":
ONNX_GS_LOGGER.colors = False
elif fmt == "timestamp":
ONNX_GS_LOGGER.timestamp = True
elif fmt == "line-info":
ONNX_GS_LOGGER.line_info = True
G_LOGGER.register_callback(set_onnx_gs_logging_level)
return self.run_impl(args)
def register_logger_callback():
from polygraphy.logger import G_LOGGER
def set_trt_logging_level(sev):
from polygraphy import mod
trt = mod.lazy_import("tensorrt")
if not mod.has_mod(trt, with_attr="__version__"):
return
if sev >= G_LOGGER.CRITICAL:
get_trt_logger().min_severity = trt.Logger.INTERNAL_ERROR
elif sev >= G_LOGGER.ERROR:
get_trt_logger().min_severity = trt.Logger.ERROR
elif sev >= G_LOGGER.INFO:
get_trt_logger().min_severity = trt.Logger.WARNING
elif sev >= G_LOGGER.VERBOSE:
get_trt_logger().min_severity = trt.Logger.INFO
else:
get_trt_logger().min_severity = trt.Logger.VERBOSE
G_LOGGER.register_callback(
set_trt_logging_level
) # Will be registered when this runner is imported.
