Revert "Bump torch-mlir to 98e08023bbf71e00ab81e980eac9f7c96f1f24b4" (#18405)
Reverts iree-org/iree#18388
https://github.com/llvm/torch-mlir/commit/fa39d91357e8ebbf375f211274dfb4bbbbbc5ccf
included in bump commit breaks some downstream usage of the IREE Python
bindings that iree-turbine depends on. Revert it while we wait on
movement upstream.
diff --git a/tests/external/iree-test-suites/onnx_ops/onnx_ops_cpu_llvm_sync.json b/tests/external/iree-test-suites/onnx_ops/onnx_ops_cpu_llvm_sync.json
index 9c7b635..3013e5d 100644
--- a/tests/external/iree-test-suites/onnx_ops/onnx_ops_cpu_llvm_sync.json
+++ b/tests/external/iree-test-suites/onnx_ops/onnx_ops_cpu_llvm_sync.json
@@ -91,6 +91,9 @@
"onnx/node/generated/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ",
"onnx/node/generated/test_castlike_FLOAT_to_FLOAT8E5M2FNUZ_expanded",
"onnx/node/generated/test_castlike_FLOAT_to_FLOAT8E5M2_expanded",
+ "onnx/node/generated/test_center_crop_pad_crop_axes_chw_expanded",
+ "onnx/node/generated/test_center_crop_pad_crop_axes_hwc_expanded",
+ "onnx/node/generated/test_center_crop_pad_crop_negative_axes_hwc_expanded",
"onnx/node/generated/test_col2im",
"onnx/node/generated/test_col2im_5d",
"onnx/node/generated/test_col2im_dilations",
@@ -100,6 +103,8 @@
"onnx/node/generated/test_compress_1",
"onnx/node/generated/test_compress_default_axis",
"onnx/node/generated/test_compress_negative_axis",
+ "onnx/node/generated/test_constant_pad_axes",
+ "onnx/node/generated/test_constant_pad_negative_axes",
"onnx/node/generated/test_conv_with_autopad_same",
"onnx/node/generated/test_convtranspose_autopad_same",
"onnx/node/generated/test_convtranspose_kernel_shape",
@@ -128,6 +133,7 @@
"onnx/node/generated/test_dft_inverse_opset19",
"onnx/node/generated/test_dft_opset19",
"onnx/node/generated/test_edge_pad",
+ "onnx/node/generated/test_einsum_batch_matmul",
"onnx/node/generated/test_einsum_sum",
"onnx/node/generated/test_gridsample_bicubic",
"onnx/node/generated/test_gridsample_bicubic_align_corners_0_additional_1",
diff --git a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_cuda.json b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_cuda.json
index fddbeb6..70c58c4 100644
--- a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_cuda.json
+++ b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_cuda.json
@@ -97,8 +97,11 @@
"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",
@@ -109,6 +112,8 @@
"onnx/node/generated/test_compress_1",
"onnx/node/generated/test_compress_default_axis",
"onnx/node/generated/test_compress_negative_axis",
+ "onnx/node/generated/test_constant_pad_axes",
+ "onnx/node/generated/test_constant_pad_negative_axes",
"onnx/node/generated/test_conv_with_autopad_same",
"onnx/node/generated/test_convtranspose_autopad_same",
"onnx/node/generated/test_convtranspose_kernel_shape",
@@ -137,6 +142,7 @@
"onnx/node/generated/test_dft_inverse_opset19",
"onnx/node/generated/test_dft_opset19",
"onnx/node/generated/test_edge_pad",
+ "onnx/node/generated/test_einsum_batch_matmul",
"onnx/node/generated/test_einsum_sum",
"onnx/node/generated/test_gridsample_bicubic",
"onnx/node/generated/test_gridsample_bicubic_align_corners_0_additional_1",
diff --git a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_rocm_rdna3.json b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_rocm_rdna3.json
index c7a3568..11a2ccf 100644
--- a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_rocm_rdna3.json
+++ b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_rocm_rdna3.json
@@ -92,8 +92,11 @@
"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",
@@ -104,6 +107,8 @@
"onnx/node/generated/test_compress_1",
"onnx/node/generated/test_compress_default_axis",
"onnx/node/generated/test_compress_negative_axis",
+ "onnx/node/generated/test_constant_pad_axes",
+ "onnx/node/generated/test_constant_pad_negative_axes",
"onnx/node/generated/test_conv_with_autopad_same",
"onnx/node/generated/test_convtranspose_autopad_same",
"onnx/node/generated/test_convtranspose_kernel_shape",
@@ -132,6 +137,7 @@
"onnx/node/generated/test_dft_inverse_opset19",
"onnx/node/generated/test_dft_opset19",
"onnx/node/generated/test_edge_pad",
+ "onnx/node/generated/test_einsum_batch_matmul",
"onnx/node/generated/test_einsum_sum",
"onnx/node/generated/test_gridsample_bicubic",
"onnx/node/generated/test_gridsample_bicubic_align_corners_0_additional_1",
diff --git a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_vulkan.json b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_vulkan.json
index ad56024..758ee8e 100644
--- a/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_vulkan.json
+++ b/tests/external/iree-test-suites/onnx_ops/onnx_ops_gpu_vulkan.json
@@ -135,6 +135,8 @@
"onnx/node/generated/test_compress_1",
"onnx/node/generated/test_compress_default_axis",
"onnx/node/generated/test_compress_negative_axis",
+ "onnx/node/generated/test_constant_pad_axes",
+ "onnx/node/generated/test_constant_pad_negative_axes",
"onnx/node/generated/test_conv_with_autopad_same",
"onnx/node/generated/test_convtranspose_autopad_same",
"onnx/node/generated/test_convtranspose_kernel_shape",
@@ -163,6 +165,7 @@
"onnx/node/generated/test_dft_inverse_opset19",
"onnx/node/generated/test_dft_opset19",
"onnx/node/generated/test_edge_pad",
+ "onnx/node/generated/test_einsum_batch_matmul",
"onnx/node/generated/test_einsum_sum",
"onnx/node/generated/test_gridsample",
"onnx/node/generated/test_gridsample_aligncorners_true",
@@ -547,7 +550,6 @@
"onnx/node/generated/test_dynamicquantizelinear",
"onnx/node/generated/test_dynamicquantizelinear_expanded",
"onnx/node/generated/test_einsum_batch_diagonal",
- "onnx/node/generated/test_einsum_batch_matmul",
"onnx/node/generated/test_einsum_transpose",
"onnx/node/generated/test_eyelike_with_dtype",
"onnx/node/generated/test_isinf_float16",
diff --git a/third_party/torch-mlir b/third_party/torch-mlir
index 98e0802..b3b8e2e 160000
--- a/third_party/torch-mlir
+++ b/third_party/torch-mlir
@@ -1 +1 @@
-Subproject commit 98e08023bbf71e00ab81e980eac9f7c96f1f24b4
+Subproject commit b3b8e2e96a6af8b9e838c07b3095b8633c701526
