diff --git a/test/test_network_ops/test_grid_sampler_2d.py b/test/test_network_ops/test_grid_sampler_2d.py index 9ea285d514a0c4a80c652d0c28d915cac455538b..eba453672825b78c8080c18f1657515fb3bf00ac 100644 --- a/test/test_network_ops/test_grid_sampler_2d.py +++ b/test/test_network_ops/test_grid_sampler_2d.py @@ -65,12 +65,11 @@ class TestGridSampler2D(TestCase): ] for item in shape_format: cpu_input, npu_input = create_common_tensor(item[0], 1, 100) - cpu_grid, npu_grid = create_common_tensor(item[1], -1, 1) + cpu_grid, npu_grid = create_common_tensor(item[1], -3, 3) cpu_output = self.cpu_op_fp16_exec(cpu_input, cpu_grid) npu_output = self.npu_op_exec(npu_input, npu_grid) self.assertRtolEqual(cpu_output, npu_output) - if __name__ == "__main__": run_tests() \ No newline at end of file diff --git a/torch_npu/csrc/aten/ops/GridSampler2dKernelNpu.cpp b/torch_npu/csrc/aten/ops/GridSampler2dKernelNpu.cpp index 674479dbe23b3eb768c83d7018e2c789e3a8118f..38e32044c8822b1177aab2c7b8fb50e23c989369 100644 --- a/torch_npu/csrc/aten/ops/GridSampler2dKernelNpu.cpp +++ b/torch_npu/csrc/aten/ops/GridSampler2dKernelNpu.cpp @@ -49,16 +49,13 @@ at::Tensor NPUNativeFunctions::grid_sampler_2d( at::Tensor result = OpPreparation::ApplyTensorWithFormat(dtypeCastOfSelf, outputSize, ACL_FORMAT_ND); - c10::SmallVectorinterMode = {"bilinear", "nearest", "bicubic"}; - c10::SmallVectorpaddingMode = {"zeros", "border", "reflection"}; - OpCommand cmd; cmd.Name("GridSampler2D") .Input(dtypeCastOfSelf) .Input(dtypeCastOfGrid) .Output(result) - .Attr("interpolation_mode", interMode[interpolation_mode]) - .Attr("padding_mode", paddingMode[padding_mode]) + .Attr("interpolation_mode", interpolation_mode) + .Attr("padding_mode", padding_mode) .Attr("align_corners", align_corners) .Run(); @@ -68,6 +65,5 @@ at::Tensor NPUNativeFunctions::grid_sampler_2d( } return result; } - } // namespace native } // namespace at_npu