Skip to content

# torch_tensorrt.logging

class torch_tensorrt.logging. Level ( value )

  Bases: `enum.Enum`

  Enum to set the minimum required logging level to print a message to stdout

  Debug = <LogLevel.DEBUG: 4>
  Error = <LogLevel.ERROR: 1>
  Graph = <LogLevel.GRAPH: 5>
  Info = <LogLevel.INFO: 3>
  InternalError = <LogLevel.INTERNAL_ERROR: 0>
  Warning = <LogLevel.WARNING: 2>

class torch_tensorrt.logging. debug

  Bases: `object`

  Context-manager to display full debug information through the logger

  Example:

  with torch_tensorrt.logging.debug():

    model_trt = torch_tensorrt.compile(model, ** spec)

class torch_tensorrt.logging. errors

  Bases: `object`

  Context-manager to limit displayed log messages to just errors and above

  Example:

  with torch_tensorrt.logging.errors():

    outputs = model_torchtrt(inputs)

torch_tensorrt.logging. get_is_colored_output_on ( ) → bool

  Get if colored output is enabled for logging

  Returns

    If colored output is one

  Return type

    bool

torch_tensorrt.logging. get_logging_prefix ( ) → str

  Get the prefix set for logging messages

  Returns

    Prefix used for logger

  Return type

    str

torch_tensorrt.logging. get_reportable_log_level ( ) → torch_tensorrt.logging.Level

  Get the level required for a message to be printed in the log

  Returns

    The enum representing the level required to print

  Return type

    torch_tensorrt.logging.Level

class torch_tensorrt.logging. graphs

  Bases: `object`

  Context-manager to display the results of intermediate lowering passes as well as full debug information through the logger

  Example:

  with torch_tensorrt.logging.graphs():

    model_trt = torch_tensorrt.compile(model, ** spec)

class torch_tensorrt.logging. info

  Bases: `object`

  Context-manager to display all info and greater severity messages

  Example:

  with torch_tensorrt.logging.info():

    model_trt = torch_tensorrt.compile(model, ** spec)

class torch_tensorrt.logging. internal_errors

  Bases: `object`

  Context-manager to limit displayed log messages to just internal errors

  Example:

  with torch_tensorrt.logging.internal_errors():

    outputs = model_torchtrt(inputs)

torch_tensorrt.logging. log ( level : torch_tensorrt.logging.Level , msg : str )

  Add a new message to the log

  Adds a new message to the log at a specified level. The message will only get printed out if Level > reportable_log_level

  Parameters

    - level ( torch_tensorrt.logging.Level ) – Severity of the message

    - msg ( str ) – Actual message text

torch_tensorrt.logging. set_is_colored_output_on ( colored_output_on : bool )

  Enable or disable color in the log output

  Parameters

    colored_output_on ( bool ) – If colored output should be enabled or not

torch_tensorrt.logging. set_logging_prefix ( prefix : str )

  Set the prefix used when logging messages

  Parameters

    prefix ( str ) – Prefix to use for logging messages

torch_tensorrt.logging. set_reportable_log_level ( level : torch_tensorrt.logging.Level )

  Set the level required for a message to be printed to the log

  Parameters

    level ( torch_tensorrt.logging.Level ) – The enum representing the level required to print

class torch_tensorrt.logging. warnings

  Bases: `object`

  Context-manager to limit displayed log messages to just warnings and above

  Example:

  with torch_tensorrt.logging.warnings():

    model_trt = torch_tensorrt.compile(model, ** spec)