changing gather signature

apbose · apbose · commit be2d5ff66099 · 2023-11-17T10:46:02.000-08:00
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,),

Original file line number	Diff line number	Diff line change
`@@ -183,7 +183,7 @@ def aten_ops_native_group_norm(`
`183`	`183`	`)`
`184`	`184`
`185`	`185`
`186`		`-@dynamo_tensorrt_converter(torch.ops.aten.gather)`
	`186`	`+@dynamo_tensorrt_converter(torch.ops.aten.gather.default)`
`187`	`187`	`@enforce_tensor_types(`
`188`	`188`	`{`
`189`	`189`	`0: (TRTTensor,),`