From adc7295af984509d38f2016afefa74b38bfca65b Mon Sep 17 00:00:00 2001 From: Evan Li Date: Thu, 24 Aug 2023 14:13:57 -0700 Subject: [PATCH] minor fix and issue #2185 --- py/torch_tensorrt/dynamo/conversion/converter_utils.py | 3 +++ py/torch_tensorrt/dynamo/conversion/impl/conv.py | 4 ++-- 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/py/torch_tensorrt/dynamo/conversion/converter_utils.py b/py/torch_tensorrt/dynamo/conversion/converter_utils.py index 72846c9007..5a757df23e 100644 --- a/py/torch_tensorrt/dynamo/conversion/converter_utils.py +++ b/py/torch_tensorrt/dynamo/conversion/converter_utils.py @@ -175,6 +175,9 @@ def extend_attr_to_tuple( """ if not isinstance(val, (tuple, list)): val = (val,) * num_elem + elif len(val) == 1: + val = (val[0],) * num_elem + if isinstance(val, list): val = tuple(val) return val diff --git a/py/torch_tensorrt/dynamo/conversion/impl/conv.py b/py/torch_tensorrt/dynamo/conversion/impl/conv.py index 78a1276882..285da2a04c 100644 --- a/py/torch_tensorrt/dynamo/conversion/impl/conv.py +++ b/py/torch_tensorrt/dynamo/conversion/impl/conv.py @@ -46,7 +46,7 @@ def convNd( ) # Process bias terms - if isinstance(bias, torch.Tensor): + if isinstance(bias, (torch.Tensor, np.ndarray)): # Transform the bias constant into a Numpy array bias = to_numpy(bias) @@ -67,7 +67,7 @@ def convNd( network, target, (weight, -1), {}, name + "_unsqueeze_weight" ) - elif isinstance(weight, torch.Tensor): + elif isinstance(weight, (torch.Tensor, np.ndarray)): # Transform the weight constant into a Numpy array weight = to_numpy(weight)