torch_tensorrt.logging

class torch_tensorrt.logging.debug

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

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

Example

with torch_tensorrt.logging.errors():
    outputs = model_torchtrt(inputs)

class torch_tensorrt.logging.graphs

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

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

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

Example

with torch_tensorrt.logging.internal_errors():
    outputs = model_torchtrt(inputs)

class torch_tensorrt.logging.warnings

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)