Import proper module

HuiGao-NV · HuiGao-NV · commit 27fd70f636dd · 2025-10-15T08:12:27.000Z
Signed-off-by: Hui Gao &lt;huig@nvidia.com&gt;
diff --git a/tensorrt_llm/_torch/memory_buffer_utils.py b/tensorrt_llm/_torch/memory_buffer_utils.py
@@ -1,3 +1,4 @@
+import contextlib
 import math
 from dataclasses import dataclass
 from typing import Optional

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,4 @@`
	`1`	`+import contextlib`
`1`	`2`	`import math`
`2`	`3`	`from dataclasses import dataclass`
`3`	`4`	`from typing import Optional`