From c04a60977361da06e2edf691ea333406ab5bef43 Mon Sep 17 00:00:00 2001 From: Chi_Liu <22491986+AmosLewis@users.noreply.github.com> Date: Tue, 14 May 2024 14:04:59 -0700 Subject: [PATCH] Integrate torch-mlir@ec6d7aa onnx.resize op (#17358) Solve iree issue: ONNX "resize" op test failures #17345 Signed-off-by: Lubo Litchev --- .../external_test_suite/onnx_cpu_llvm_sync.json | 14 ++++++-------- .../external_test_suite/onnx_gpu_cuda.json | 17 ++++++++--------- third_party/torch-mlir | 2 +- 3 files changed, 15 insertions(+), 18 deletions(-) diff --git a/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json b/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json index 287bba0efad7a..dd8bfbd699b15 100644 --- a/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json +++ b/build_tools/pkgci/external_test_suite/onnx_cpu_llvm_sync.json @@ -384,15 +384,11 @@ "test_resize_downsample_scales_cubic_A_n0p5_exclude_outside", "test_resize_downsample_scales_cubic_align_corners", "test_resize_downsample_scales_cubic_antialias", - "test_resize_downsample_scales_linear", - "test_resize_downsample_scales_linear_align_corners", "test_resize_downsample_scales_linear_antialias", - "test_resize_downsample_scales_linear_half_pixel_symmetric", "test_resize_downsample_scales_nearest", "test_resize_downsample_sizes_cubic", "test_resize_downsample_sizes_cubic_antialias", "test_resize_downsample_sizes_linear_antialias", - "test_resize_downsample_sizes_linear_pytorch_half_pixel", "test_resize_downsample_sizes_nearest", "test_resize_downsample_sizes_nearest_not_larger", "test_resize_downsample_sizes_nearest_not_smaller", @@ -403,9 +399,6 @@ "test_resize_upsample_scales_cubic_A_n0p5_exclude_outside", "test_resize_upsample_scales_cubic_align_corners", "test_resize_upsample_scales_cubic_asymmetric", - "test_resize_upsample_scales_linear", - "test_resize_upsample_scales_linear_align_corners", - "test_resize_upsample_scales_linear_half_pixel_symmetric", "test_resize_upsample_scales_nearest", "test_resize_upsample_scales_nearest_axes_2_3", "test_resize_upsample_scales_nearest_axes_3_2", @@ -414,7 +407,6 @@ "test_resize_upsample_sizes_nearest_axes_2_3", "test_resize_upsample_sizes_nearest_axes_3_2", "test_resize_upsample_sizes_nearest_ceil_half_pixel", - "test_resize_upsample_sizes_nearest_floor_align_corners", "test_resize_upsample_sizes_nearest_not_larger", "test_resize_upsample_sizes_nearest_round_prefer_ceil_asymmetric", "test_reversesequence_batch", @@ -629,6 +621,12 @@ "test_qlinearmatmul_3D_uint8_float32", "test_reduce_min_empty_set", "test_reduce_sum_empty_set_non_reduced_axis_zero", + "test_resize_downsample_scales_linear", + "test_resize_downsample_scales_linear_half_pixel_symmetric", + "test_resize_downsample_sizes_linear_pytorch_half_pixel", + "test_resize_upsample_scales_linear", + "test_resize_upsample_scales_linear_half_pixel_symmetric", + "test_resize_upsample_sizes_nearest_floor_align_corners", "test_scatter_elements_with_negative_indices", "test_sce_mean_no_weight_ii", "test_sce_mean_no_weight_ii_log_prob", diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json index 433952f5c7cba..23e854c23a2a5 100644 --- a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json +++ b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json @@ -12,7 +12,9 @@ "skip_run_tests": [ "test_gather_elements_negative_indices", "test_gridsample_zeros_padding", - "test_scatter_elements_with_negative_indices" + "test_scatter_elements_with_negative_indices", + "test_resize_downsample_scales_linear_align_corners", + "test_resize_downsample_scales_linear_half_pixel_symmetric" ], "expected_compile_failures": [ "test_adagrad", @@ -385,15 +387,11 @@ "test_resize_downsample_scales_cubic_A_n0p5_exclude_outside", "test_resize_downsample_scales_cubic_align_corners", "test_resize_downsample_scales_cubic_antialias", - "test_resize_downsample_scales_linear", - "test_resize_downsample_scales_linear_align_corners", "test_resize_downsample_scales_linear_antialias", - "test_resize_downsample_scales_linear_half_pixel_symmetric", "test_resize_downsample_scales_nearest", "test_resize_downsample_sizes_cubic", "test_resize_downsample_sizes_cubic_antialias", "test_resize_downsample_sizes_linear_antialias", - "test_resize_downsample_sizes_linear_pytorch_half_pixel", "test_resize_downsample_sizes_nearest", "test_resize_downsample_sizes_nearest_not_larger", "test_resize_downsample_sizes_nearest_not_smaller", @@ -404,9 +402,6 @@ "test_resize_upsample_scales_cubic_A_n0p5_exclude_outside", "test_resize_upsample_scales_cubic_align_corners", "test_resize_upsample_scales_cubic_asymmetric", - "test_resize_upsample_scales_linear", - "test_resize_upsample_scales_linear_align_corners", - "test_resize_upsample_scales_linear_half_pixel_symmetric", "test_resize_upsample_scales_nearest", "test_resize_upsample_scales_nearest_axes_2_3", "test_resize_upsample_scales_nearest_axes_3_2", @@ -415,7 +410,6 @@ "test_resize_upsample_sizes_nearest_axes_2_3", "test_resize_upsample_sizes_nearest_axes_3_2", "test_resize_upsample_sizes_nearest_ceil_half_pixel", - "test_resize_upsample_sizes_nearest_floor_align_corners", "test_resize_upsample_sizes_nearest_not_larger", "test_resize_upsample_sizes_nearest_round_prefer_ceil_asymmetric", "test_reversesequence_batch", @@ -646,6 +640,11 @@ "test_reduce_sum_square_default_axes_keepdims_example_expanded", "test_reduce_sum_square_default_axes_keepdims_random", "test_reduce_sum_square_default_axes_keepdims_random_expanded", + "test_resize_downsample_scales_linear", + "test_resize_downsample_sizes_linear_pytorch_half_pixel", + "test_resize_upsample_scales_linear", + "test_resize_upsample_scales_linear_half_pixel_symmetric", + "test_resize_upsample_sizes_nearest_floor_align_corners", "test_sce_mean_no_weight_ii", "test_sce_mean_no_weight_ii_log_prob", "test_shape_end_1", diff --git a/third_party/torch-mlir b/third_party/torch-mlir index bce800a3f4200..ec6d7aa5d28f1 160000 --- a/third_party/torch-mlir +++ b/third_party/torch-mlir @@ -1 +1 @@ -Subproject commit bce800a3f4200f8a28e844a91c2be2d2ce71c174 +Subproject commit ec6d7aa5d28f110aa5b893e16e502e6198988801