Integrate torch-mlir@ec6d7aa onnx.resize op (iree-org#17358)

Solve iree issue: ONNX "resize" op test failures iree-org#17345 Signed-off-by: Lubo Litchev <[email protected]>
LLITCHEV · Jul 30, 2024 · c04a609 · c04a609
1 parent c5b7407
commit c04a609
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 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
@@ -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
@@ -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