diff --git a/py/torch_tensorrt/dynamo/runtime/_ResourceAllocator.py b/py/torch_tensorrt/dynamo/runtime/_ResourceAllocator.py index 3e570f4d78..211bbbb068 100644 --- a/py/torch_tensorrt/dynamo/runtime/_ResourceAllocator.py +++ b/py/torch_tensorrt/dynamo/runtime/_ResourceAllocator.py @@ -21,7 +21,6 @@ def __init__( self.dynamically_allocate_resources = dynamically_allocate_resources def __enter__(self) -> None: - print("Entering resource allocator context") for name, submodule in self.compiled_module.named_modules(): if "_run_on_acc" in name: submodule.use_dynamically_allocated_resources( @@ -32,5 +31,5 @@ def __exit__(self, exc_type: Any, exc_value: Any, exc_tb: Any) -> None: for name, submodule in self.compiled_module.named_modules(): if "_run_on_acc" in name: submodule.use_dynamically_allocated_resources( - dynamically_allocate_resources=self.dynamically_allocate_resources + dynamically_allocate_resources=False )