changing gather signature

pytorch · Nov 17, 2023 · be2d5ff · be2d5ff
1 parent c4ae041
commit be2d5ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,),