diff --git a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py index 45949a1c8d..915ac8af4d 100644 --- a/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py +++ b/py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py @@ -330,8 +330,8 @@ 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) +@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler.default) +@dynamo_tensorrt_converter(torch.ops.aten.grid_sampler_2d.default) @enforce_tensor_types( { 0: (TRTTensor,),