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 2b845a5058b2..31e6a30d230b 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
@@ -171,7 +171,6 @@
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum",
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii",
-    "onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight",
     "onnx/node/generated/test_nonmaxsuppression_center_point_box_format",
@@ -334,10 +333,6 @@
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob",
@@ -358,9 +353,6 @@
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded",
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight",
-    "onnx/node/generated/test_sce_mean_weight_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii",
     "onnx/node/generated/test_sce_mean_weight_ii_3d",
     "onnx/node/generated/test_sce_mean_weight_ii_3d_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob",
@@ -369,15 +361,6 @@
     "onnx/node/generated/test_sce_mean_weight_ii_4d_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_log_prob_expanded",
-    "onnx/node/generated/test_sce_none_weights",
-    "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_slice",
     "onnx/node/generated/test_slice_default_steps",
     "onnx/node/generated/test_slice_end_out_of_bounds",
@@ -438,6 +421,10 @@
     "onnx/node/generated/test_reduce_min_empty_set",
     "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_sce_mean_weight",
+    "onnx/node/generated/test_sce_mean_weight_ii",
+    "onnx/node/generated/test_sce_mean_weight_ii_log_prob",
+    "onnx/node/generated/test_sce_mean_weight_log_prob",
     "onnx/node/generated/test_shape_end_1",
     "onnx/node/generated/test_shape_start_1_end_2",
     "onnx/node/generated/test_shape_start_1_end_negative_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 11beff692857..c4c0ba917e78 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
@@ -174,7 +174,6 @@
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum",
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii",
-    "onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight",
     "onnx/node/generated/test_nonmaxsuppression_center_point_box_format",
@@ -335,10 +334,6 @@
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob",
@@ -359,9 +354,6 @@
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_expanded",
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight",
-    "onnx/node/generated/test_sce_mean_weight_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii",
     "onnx/node/generated/test_sce_mean_weight_ii_3d",
     "onnx/node/generated/test_sce_mean_weight_ii_3d_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_3d_log_prob",
@@ -370,15 +362,6 @@
     "onnx/node/generated/test_sce_mean_weight_ii_4d_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_log_prob_expanded",
-    "onnx/node/generated/test_sce_none_weights",
-    "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_slice",
     "onnx/node/generated/test_slice_default_steps",
     "onnx/node/generated/test_slice_end_out_of_bounds",
@@ -485,6 +468,10 @@
     "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_resize_downsample_scales_linear_align_corners",
+    "onnx/node/generated/test_sce_mean_weight",
+    "onnx/node/generated/test_sce_mean_weight_ii",
+    "onnx/node/generated/test_sce_mean_weight_ii_log_prob",
+    "onnx/node/generated/test_sce_mean_weight_log_prob",
     "onnx/node/generated/test_shape",
     "onnx/node/generated/test_shape_clip_end",
     "onnx/node/generated/test_shape_clip_start",
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 eafe6786fd44..65d75dc413b7 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
@@ -219,7 +219,6 @@
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum",
     "onnx/node/generated/test_nllloss_NCd1d2_with_weight_reduction_sum_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3_none_no_weight_negative_ii",
-    "onnx/node/generated/test_nllloss_NCd1d2d3_sum_weight_high_ii",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_nllloss_NCd1d2d3d4d5_none_no_weight",
     "onnx/node/generated/test_nonmaxsuppression_center_point_box_format",
@@ -399,10 +398,6 @@
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob",
     "onnx/node/generated/test_sce_NCd1d2d3_none_no_weight_negative_ii_log_prob_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_expanded",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob",
-    "onnx/node/generated/test_sce_NCd1d2d3_sum_weight_high_ii_log_prob_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_expanded",
     "onnx/node/generated/test_sce_NCd1d2d3d4d5_mean_weight_log_prob",
@@ -424,7 +419,6 @@
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_no_weight_ii_4d_log_prob_expanded",
     "onnx/node/generated/test_sce_mean_weight",
-    "onnx/node/generated/test_sce_mean_weight_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii",
     "onnx/node/generated/test_sce_mean_weight_ii_3d",
     "onnx/node/generated/test_sce_mean_weight_ii_3d_expanded",
@@ -434,15 +428,8 @@
     "onnx/node/generated/test_sce_mean_weight_ii_4d_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob",
     "onnx/node/generated/test_sce_mean_weight_ii_4d_log_prob_expanded",
-    "onnx/node/generated/test_sce_mean_weight_ii_expanded",
     "onnx/node/generated/test_sce_mean_weight_ii_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_ii_log_prob_expanded",
     "onnx/node/generated/test_sce_mean_weight_log_prob",
-    "onnx/node/generated/test_sce_mean_weight_log_prob_expanded",
-    "onnx/node/generated/test_sce_none_weights",
-    "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_slice",
     "onnx/node/generated/test_slice_default_steps",
     "onnx/node/generated/test_slice_end_out_of_bounds",
diff --git a/third_party/torch-mlir b/third_party/torch-mlir
index 6aa46967b69a..8eb34dae7894 160000
--- a/third_party/torch-mlir
+++ b/third_party/torch-mlir
@@ -1 +1 @@
-Subproject commit 6aa46967b69a01a46d56146250978d08e243e75e
+Subproject commit 8eb34dae78940efe529fedef5bbe96c905f3ee3b