Shortcuts

torch_tensorrt.logging

class torch_tensorrt.logging.debug[source]

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[source]

Bases: object

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[source]

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[source]

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[source]

Bases: object

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[source]

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)

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources