changing gather signature

Author: apbose

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,),

py/torch_tensorrt/dynamo/conversion/impl/select.py

@@ -68,7 +68,7 @@ def select(
     indices_tensor = ctx.net.add_constant(
         index_value.shape, to_numpy(index_value)
     ).get_output(0)
-    out = gather(input, indices_tensor, dim)
+    out = gather(ctx, target, source_ir, name, input, indices_tensor, dim)
     if len(out.shape) != 1:
         layer = ctx.net.add_shuffle(out)
     return layer.get_output(0)
@@ -140,7 +140,7 @@ def index(
         )
         index = adv_indx_indices[0]
         _LOGGER.debug(f"The advanced index indices is {adv_indx_indices}")
-        return gather(input, index, indices_tensor)
+        return gather(ctx, target, source_ir, name, input, index, indices_tensor)
     else:
         input_shape = input.shape
         _LOGGER.debug(f"The input shape is {input.shape}")
@@ -253,7 +253,7 @@ def index(
                     dim_tensor_list[adv_indx_indices[i]],
                 )
 
-        gather_out = gather(flatten_tensor, cum_adv_index, 0)
+        gather_out = gather(ctx, target, source_ir, name, flatten_tensor, 0, cum_adv_index)
         _LOGGER.debug(f"The shape after cumultative gather is {gather_out.shape}")
         _LOGGER.debug(f"The shape for cumulative adv index is {cum_adv_index}")