Keeping the ignore[misc] and changing converter key

Author: apbose

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

@@ -330,11 +330,7 @@ def aten_ops_fmod(
     return impl.elementwise.fmod(ctx, target, SourceIR.ATEN, name, args[0], args[1])
 
 
-@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler)
-@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_2d.out)
-@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_2d_backward.out)
-@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_3d.out)
-@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_3d_backward.out)
+@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler)  # type: ignore[misc]
 @enforce_tensor_types(
     {
         0: (TRTTensor,),
@@ -349,21 +345,19 @@ def aten_ops_grid(
     name: str,
 ) -> Union[TRTTensor, Sequence[TRTTensor]]:
     return impl.grid.grid(
-        ctx, 
-        target, 
-        SourceIR.ATEN, 
-        name, 
-        input=args[0], 
-        grid=args[1], 
-        interpolation_mode=args[2], 
-        padding_mode=args[3], 
+        ctx,
+        target,
+        SourceIR.ATEN,
+        name,
+        input=args[0],
+        grid=args[1],
+        interpolation_mode=args[2],
+        padding_mode=args[3],
         align_corners=args_bounds_check(args, 4, True),
-        output_mask=args_bounds_check(args, 5, None),
-
     )
 
 
-@dynamo_tensorrt_converter(torch.ops.aten.relu.default)
+@dynamo_tensorrt_converter(torch.ops.aten.relu.default)  # type: ignore[misc]
 def aten_ops_relu(
     ctx: ConversionContext,
     target: Target,

py/torch_tensorrt/dynamo/conversion/impl/grid.py

@@ -34,7 +34,6 @@ def grid(
     interpolation_mode: int,
     padding_mode: int,
     align_corners: bool,
-    output_mask: Optional[Sequence[bool]] = None,
 ) -> TRTTensor:
     grid_layer = ctx.net.add_grid_sample(input, grid)
     assert interpolation_mode in GridSamplerInterpolationMode
@@ -45,11 +44,4 @@ def grid(
     grid_layer.sample_mode = GridSamplerSampling.get(padding_mode, None)
     grid_layer.align_corners = align_corners
     set_layer_name(grid_layer, target, name + "_grid_layer", source_ir)
-    if output_mask is None:
-        return grid_layer.get_output(0)
-    elif output_mask[0] and output_mask[1]:
-        return (grid_layer.get_output(0), None)
-    elif output_mask[0]:
-        return grid_layer.get_output(0)
-    else:
-        return None
+    return grid_layer.get_output(0)