Skip to content

Commit 45e9184

Browse files
committed
fix: Upgrade to_numpy to allow boolean constants
1 parent efc5bd7 commit 45e9184

File tree

8 files changed

+186
-126
lines changed

8 files changed

+186
-126
lines changed

py/torch_tensorrt/dynamo/conversion/_TRTInterpreter.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -345,7 +345,7 @@ def call_function(self, target: str, args: Any, kwargs: Any) -> Any:
345345

346346
def get_attr(self, target: str, args: Any, kwargs: Any) -> np.ndarray:
347347
with _disable_current_modes():
348-
from torch_tensorrt.fx.converters import to_numpy
348+
from torch_tensorrt.dynamo.conversion.converter_utils import to_numpy
349349

350350
frozen_attr = self.fetch_attr(target)
351351

0 commit comments

Comments
 (0)