From be2d5ff6609937d0e46025059d0f073478299387 Mon Sep 17 00:00:00 2001 From: apbose Date: Fri, 17 Nov 2023 10:46:02 -0800 Subject: [PATCH] changing gather signature --- py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py index 27b90aaa37..cd0bb5831e 100644 --- a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py +++ b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py @@ -183,7 +183,7 @@ def aten_ops_native_group_norm( ) -@dynamo_tensorrt_converter(torch.ops.aten.gather) +@dynamo_tensorrt_converter(torch.ops.aten.gather.default) @enforce_tensor_types( { 0: (TRTTensor,),