Bump torch-mlir to 8358e8c255cb5945b0f69c5fd0c5c5f738cea3f4 (#18169)

Bump torch-mlir to llvm/torch-mlir@8358e8c --------- Signed-off-by: PhaneeshB <[email protected]> Co-authored-by: Scott Todd <[email protected]>
iree-org · Aug 10, 2024 · e36aa78 · e36aa78
1 parent 5a48912
commit e36aa78
Show file tree

Hide file tree

Showing 5 changed files with 9 additions and 71 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
@@ -99,15 +99,9 @@
     "onnx/node/generated/test_castlike_FLOAT_to_STRING_expanded",
     "onnx/node/generated/test_castlike_STRING_to_FLOAT",
     "onnx/node/generated/test_castlike_STRING_to_FLOAT_expanded",
-    "onnx/node/generated/test_center_crop_pad_crop",
-    "onnx/node/generated/test_center_crop_pad_crop_and_pad",
-    "onnx/node/generated/test_center_crop_pad_crop_axes_chw",
     "onnx/node/generated/test_center_crop_pad_crop_axes_chw_expanded",
-    "onnx/node/generated/test_center_crop_pad_crop_axes_hwc",
     "onnx/node/generated/test_center_crop_pad_crop_axes_hwc_expanded",
-    "onnx/node/generated/test_center_crop_pad_crop_negative_axes_hwc",
     "onnx/node/generated/test_center_crop_pad_crop_negative_axes_hwc_expanded",
-    "onnx/node/generated/test_center_crop_pad_pad",
     "onnx/node/generated/test_col2im",
     "onnx/node/generated/test_col2im_5d",
     "onnx/node/generated/test_col2im_dilations",
@@ -168,9 +162,7 @@
     "onnx/node/generated/test_gridsample_volumetric_nearest_align_corners_0",
     "onnx/node/generated/test_gridsample_volumetric_nearest_align_corners_1",
     "onnx/node/generated/test_group_normalization_epsilon",
-    "onnx/node/generated/test_group_normalization_epsilon_expanded",
     "onnx/node/generated/test_group_normalization_example",
-    "onnx/node/generated/test_group_normalization_example_expanded",
     "onnx/node/generated/test_gru_batchwise",
     "onnx/node/generated/test_gru_defaults",
     "onnx/node/generated/test_gru_seq_length",
@@ -204,7 +196,6 @@
     "onnx/node/generated/test_maxpool_2d_precomputed_same_upper",
     "onnx/node/generated/test_maxpool_2d_same_lower",
     "onnx/node/generated/test_maxpool_2d_same_upper",
-    "onnx/node/generated/test_maxpool_2d_uint8",
     "onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides",
     "onnx/node/generated/test_maxunpool_export_with_output_shape",
     "onnx/node/generated/test_maxunpool_export_without_output_shape",
@@ -242,8 +233,6 @@
     "onnx/node/generated/test_nonmaxsuppression_two_batches",
     "onnx/node/generated/test_nonmaxsuppression_two_classes",
     "onnx/node/generated/test_nonzero_example",
-    "onnx/node/generated/test_onehot_with_axis",
-    "onnx/node/generated/test_onehot_with_negative_axis",
     "onnx/node/generated/test_pow_types_int32_float32",
     "onnx/node/generated/test_pow_types_int32_int32",
     "onnx/node/generated/test_pow_types_int64_float32",
@@ -395,8 +384,6 @@
     "onnx/node/generated/test_roialign_aligned_false",
     "onnx/node/generated/test_roialign_aligned_true",
     "onnx/node/generated/test_roialign_mode_max",
-    "onnx/node/generated/test_scan9_sum",
-    "onnx/node/generated/test_scan_sum",
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
@@ -452,8 +439,7 @@
     "onnx/node/generated/test_sce_none_weights_expanded",
     "onnx/node/generated/test_sce_none_weights_log_prob",
     "onnx/node/generated/test_sce_none_weights_log_prob_expanded",
-    "onnx/node/generated/test_shrink_hard",
-    "onnx/node/generated/test_shrink_soft",
+    "onnx/node/generated/test_shape_end_negative_1",
     "onnx/node/generated/test_simple_rnn_batchwise",
     "onnx/node/generated/test_simple_rnn_defaults",
     "onnx/node/generated/test_simple_rnn_with_initial_bias",
@@ -464,14 +450,6 @@
     "onnx/node/generated/test_slice_neg_steps",
     "onnx/node/generated/test_slice_negative_axes",
     "onnx/node/generated/test_slice_start_out_of_bounds",
-    "onnx/node/generated/test_split_equal_parts_1d_opset13",
-    "onnx/node/generated/test_split_equal_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset18",
-    "onnx/node/generated/test_split_variable_parts_2d_opset13",
-    "onnx/node/generated/test_split_variable_parts_2d_opset18",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset18",
     "onnx/node/generated/test_stft",
     "onnx/node/generated/test_stft_with_window",
     "onnx/node/generated/test_string_concat",
@@ -527,7 +505,6 @@
     "onnx/node/generated/test_constantofshape_int_shape_zero",
     "onnx/node/generated/test_constantofshape_int_zeros",
     "onnx/node/generated/test_dropout_default_mask_ratio",
-    "onnx/node/generated/test_gather_elements_negative_indices",
     "onnx/node/generated/test_gridsample_nearest",
     "onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
     "onnx/node/generated/test_gridsample_nearest_align_corners_1_additional_1",
@@ -553,6 +530,7 @@
     "onnx/node/generated/test_reduce_sum_empty_axes_input_noop",
     "onnx/node/generated/test_reduce_sum_empty_set_non_reduced_axis_zero",
     "onnx/node/generated/test_resize_downsample_scales_linear_align_corners",
+    "onnx/node/generated/test_shape_clip_start",
     "onnx/node/generated/test_shape_end_1",
     "onnx/node/generated/test_shape_start_1",
     "onnx/node/generated/test_shape_start_1_end_2",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
@@ -207,7 +207,6 @@
     "onnx/node/generated/test_maxpool_2d_precomputed_same_upper",
     "onnx/node/generated/test_maxpool_2d_same_lower",
     "onnx/node/generated/test_maxpool_2d_same_upper",
-    "onnx/node/generated/test_maxpool_2d_uint8",
     "onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides",
     "onnx/node/generated/test_maxunpool_export_with_output_shape",
     "onnx/node/generated/test_maxunpool_export_without_output_shape",
@@ -242,8 +241,6 @@
     "onnx/node/generated/test_nonmaxsuppression_two_batches",
     "onnx/node/generated/test_nonmaxsuppression_two_classes",
     "onnx/node/generated/test_nonzero_example",
-    "onnx/node/generated/test_onehot_with_axis",
-    "onnx/node/generated/test_onehot_with_negative_axis",
     "onnx/node/generated/test_pow_types_int32_float32",
     "onnx/node/generated/test_pow_types_int32_int32",
     "onnx/node/generated/test_pow_types_int64_float32",
@@ -393,8 +390,6 @@
     "onnx/node/generated/test_roialign_aligned_false",
     "onnx/node/generated/test_roialign_aligned_true",
     "onnx/node/generated/test_roialign_mode_max",
-    "onnx/node/generated/test_scan9_sum",
-    "onnx/node/generated/test_scan_sum",
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
@@ -450,8 +445,7 @@
     "onnx/node/generated/test_sce_none_weights_expanded",
     "onnx/node/generated/test_sce_none_weights_log_prob",
     "onnx/node/generated/test_sce_none_weights_log_prob_expanded",
-    "onnx/node/generated/test_shrink_hard",
-    "onnx/node/generated/test_shrink_soft",
+    "onnx/node/generated/test_shape_end_negative_1",
     "onnx/node/generated/test_simple_rnn_batchwise",
     "onnx/node/generated/test_simple_rnn_defaults",
     "onnx/node/generated/test_simple_rnn_with_initial_bias",
@@ -462,14 +456,6 @@
     "onnx/node/generated/test_slice_neg_steps",
     "onnx/node/generated/test_slice_negative_axes",
     "onnx/node/generated/test_slice_start_out_of_bounds",
-    "onnx/node/generated/test_split_equal_parts_1d_opset13",
-    "onnx/node/generated/test_split_equal_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset18",
-    "onnx/node/generated/test_split_variable_parts_2d_opset13",
-    "onnx/node/generated/test_split_variable_parts_2d_opset18",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset18",
     "onnx/node/generated/test_stft",
     "onnx/node/generated/test_stft_with_window",
     "onnx/node/generated/test_string_concat",
@@ -566,6 +552,7 @@
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_example_expanded",
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random",
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random_expanded",
+    "onnx/node/generated/test_shape_clip_start",
     "onnx/node/generated/test_shape_end_1",
     "onnx/node/generated/test_shape_start_1",
     "onnx/node/generated/test_shape_start_1_end_2",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json b/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
@@ -202,7 +202,6 @@
     "onnx/node/generated/test_maxpool_2d_precomputed_same_upper",
     "onnx/node/generated/test_maxpool_2d_same_lower",
     "onnx/node/generated/test_maxpool_2d_same_upper",
-    "onnx/node/generated/test_maxpool_2d_uint8",
     "onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides",
     "onnx/node/generated/test_maxunpool_export_with_output_shape",
     "onnx/node/generated/test_maxunpool_export_without_output_shape",
@@ -237,8 +236,6 @@
     "onnx/node/generated/test_nonmaxsuppression_two_batches",
     "onnx/node/generated/test_nonmaxsuppression_two_classes",
     "onnx/node/generated/test_nonzero_example",
-    "onnx/node/generated/test_onehot_with_axis",
-    "onnx/node/generated/test_onehot_with_negative_axis",
     "onnx/node/generated/test_pow_types_int32_float32",
     "onnx/node/generated/test_pow_types_int32_int32",
     "onnx/node/generated/test_pow_types_int64_float32",
@@ -388,8 +385,6 @@
     "onnx/node/generated/test_roialign_aligned_false",
     "onnx/node/generated/test_roialign_aligned_true",
     "onnx/node/generated/test_roialign_mode_max",
-    "onnx/node/generated/test_scan9_sum",
-    "onnx/node/generated/test_scan_sum",
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
@@ -445,8 +440,7 @@
     "onnx/node/generated/test_sce_none_weights_expanded",
     "onnx/node/generated/test_sce_none_weights_log_prob",
     "onnx/node/generated/test_sce_none_weights_log_prob_expanded",
-    "onnx/node/generated/test_shrink_hard",
-    "onnx/node/generated/test_shrink_soft",
+    "onnx/node/generated/test_shape_end_negative_1",
     "onnx/node/generated/test_simple_rnn_batchwise",
     "onnx/node/generated/test_simple_rnn_defaults",
     "onnx/node/generated/test_simple_rnn_with_initial_bias",
@@ -457,14 +451,6 @@
     "onnx/node/generated/test_slice_neg_steps",
     "onnx/node/generated/test_slice_negative_axes",
     "onnx/node/generated/test_slice_start_out_of_bounds",
-    "onnx/node/generated/test_split_equal_parts_1d_opset13",
-    "onnx/node/generated/test_split_equal_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset18",
-    "onnx/node/generated/test_split_variable_parts_2d_opset13",
-    "onnx/node/generated/test_split_variable_parts_2d_opset18",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset18",
     "onnx/node/generated/test_stft",
     "onnx/node/generated/test_stft_with_window",
     "onnx/node/generated/test_string_concat",
@@ -539,7 +525,6 @@
     "onnx/node/generated/test_eyelike_populate_off_main_diagonal",
     "onnx/node/generated/test_eyelike_with_dtype",
     "onnx/node/generated/test_eyelike_without_dtype",
-    "onnx/node/generated/test_gather_elements_negative_indices",
     "onnx/node/generated/test_gridsample_nearest",
     "onnx/node/generated/test_gridsample_nearest_align_corners_0_additional_1",
     "onnx/node/generated/test_gridsample_nearest_align_corners_1_additional_1",
@@ -592,7 +577,6 @@
     "onnx/node/generated/test_shape_clip_end",
     "onnx/node/generated/test_shape_clip_start",
     "onnx/node/generated/test_shape_end_1",
-    "onnx/node/generated/test_shape_end_negative_1",
     "onnx/node/generated/test_shape_example",
     "onnx/node/generated/test_shape_start_1",
     "onnx/node/generated/test_shape_start_1_end_2",

diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json b/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
@@ -240,7 +240,6 @@
     "onnx/node/generated/test_maxpool_2d_precomputed_same_upper",
     "onnx/node/generated/test_maxpool_2d_same_lower",
     "onnx/node/generated/test_maxpool_2d_same_upper",
-    "onnx/node/generated/test_maxpool_2d_uint8",
     "onnx/node/generated/test_maxpool_with_argmax_2d_precomputed_strides",
     "onnx/node/generated/test_maxunpool_export_with_output_shape",
     "onnx/node/generated/test_maxunpool_export_without_output_shape",
@@ -449,8 +448,6 @@
     "onnx/node/generated/test_roialign_aligned_false",
     "onnx/node/generated/test_roialign_aligned_true",
     "onnx/node/generated/test_roialign_mode_max",
-    "onnx/node/generated/test_scan9_sum",
-    "onnx/node/generated/test_scan_sum",
     "onnx/node/generated/test_scatter_elements_with_duplicate_indices",
     "onnx/node/generated/test_scatter_elements_with_reduction_max",
     "onnx/node/generated/test_scatter_elements_with_reduction_min",
@@ -506,8 +503,7 @@
     "onnx/node/generated/test_sce_none_weights_expanded",
     "onnx/node/generated/test_sce_none_weights_log_prob",
     "onnx/node/generated/test_sce_none_weights_log_prob_expanded",
-    "onnx/node/generated/test_shrink_hard",
-    "onnx/node/generated/test_shrink_soft",
+    "onnx/node/generated/test_shape_end_negative_1",
     "onnx/node/generated/test_simple_rnn_batchwise",
     "onnx/node/generated/test_simple_rnn_defaults",
     "onnx/node/generated/test_simple_rnn_with_initial_bias",
@@ -518,14 +514,6 @@
     "onnx/node/generated/test_slice_neg_steps",
     "onnx/node/generated/test_slice_negative_axes",
     "onnx/node/generated/test_slice_start_out_of_bounds",
-    "onnx/node/generated/test_split_equal_parts_1d_opset13",
-    "onnx/node/generated/test_split_equal_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset13",
-    "onnx/node/generated/test_split_variable_parts_1d_opset18",
-    "onnx/node/generated/test_split_variable_parts_2d_opset13",
-    "onnx/node/generated/test_split_variable_parts_2d_opset18",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset13",
-    "onnx/node/generated/test_split_variable_parts_default_axis_opset18",
     "onnx/node/generated/test_stft",
     "onnx/node/generated/test_stft_with_window",
     "onnx/node/generated/test_string_concat",
@@ -621,14 +609,14 @@
     "onnx/node/generated/test_dynamicquantizelinear_expanded",
     "onnx/node/generated/test_einsum_transpose",
     "onnx/node/generated/test_eyelike_with_dtype",
-    "onnx/node/generated/test_gather_elements_negative_indices",
     "onnx/node/generated/test_isinf_float16",
     "onnx/node/generated/test_isnan_float16",
     "onnx/node/generated/test_matmulinteger",
     "onnx/node/generated/test_max_float16",
     "onnx/node/generated/test_max_float64",
     "onnx/node/generated/test_max_int16",
     "onnx/node/generated/test_max_int8",
+    "onnx/node/generated/test_maxpool_2d_uint8",
     "onnx/node/generated/test_min_float16",
     "onnx/node/generated/test_min_float64",
     "onnx/node/generated/test_min_int16",
@@ -680,6 +668,7 @@
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_example_expanded",
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random",
     "onnx/node/generated/test_reduce_sum_square_default_axes_keepdims_random_expanded",
+    "onnx/node/generated/test_shape_clip_start",
     "onnx/node/generated/test_shape_end_1",
     "onnx/node/generated/test_shape_start_1",
     "onnx/node/generated/test_shape_start_1_end_2",

diff --git a/third_party/torch-mlir b/third_party/torch-mlir
+1 −1		externals/llvm-project
+6 −0		include/torch-mlir/Conversion/TorchOnnxToTorch/Utils.h
+53 −19		lib/Conversion/TorchOnnxToTorch/DefaultDomainAtoF.cpp
+23 −6		lib/Conversion/TorchOnnxToTorch/DefaultDomainGtoP.cpp
+169 −20		lib/Conversion/TorchOnnxToTorch/DefaultDomainQtoZ.cpp
+28 −3		lib/Conversion/TorchToArith/TorchToArith.cpp
+18 −12		lib/Conversion/TorchToLinalg/Pooling.cpp
+23 −0		lib/Conversion/Utils/Utils.cpp
+12 −1		lib/Dialect/Torch/IR/TorchOps.cpp
+12 −9		lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp
+9 −0		lib/Dialect/Torch/Transforms/FuseQuantizedOps.cpp
+16 −1		projects/pt1/e2e_testing/xfail_sets.py
+8 −1		projects/pt1/python/torch_mlir_e2e_test/framework.py
+62 −0		projects/pt1/python/torch_mlir_e2e_test/test_suite/basic.py
+24 −0		projects/pt1/python/torch_mlir_e2e_test/test_suite/scalar.py
+2 −0		python/torch_mlir/extras/fx_importer.py
+1 −1		pytorch-hash.txt
+1 −1		pytorch-requirements.txt
+46 −30		test/Conversion/TorchOnnxToTorch/simple_ops_a_to_f.mlir
+17 −9		test/Conversion/TorchOnnxToTorch/simple_ops_g_to_p.mlir
+104 −13		test/Conversion/TorchOnnxToTorch/simple_ops_q_to_z.mlir
+12 −0		test/Dialect/Torch/canonicalize.mlir
+1 −1		torchvision-requirements.txt