Skip to content

Commit be84ba5

Browse files
committed
fix: Upgrade to_numpy to allow boolean constants
1 parent 2ff5466 commit be84ba5

File tree

9 files changed

+185
-128
lines changed

9 files changed

+185
-128
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)