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 6b27dbfdc04d..8f9615459ab5 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
@@ -44,7 +44,6 @@
     "test_basic_deform_conv_without_padding",
     "test_batchnorm_epsilon_training_mode",
     "test_batchnorm_example_training_mode",
-    "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
@@ -544,6 +543,7 @@
     "test_asin",
     "test_averagepool_2d_ceil",
     "test_bernoulli",
+    "test_bernoulli_double",
     "test_bernoulli_double_expanded",
     "test_bernoulli_expanded",
     "test_cast_BFLOAT16_to_FLOAT",
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 4dde787e20e7..5f1d3b518d9c 100644
--- a/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
+++ b/build_tools/pkgci/external_test_suite/onnx_gpu_cuda.json
@@ -50,7 +50,6 @@
     "test_basic_deform_conv_without_padding",
     "test_batchnorm_epsilon_training_mode",
     "test_batchnorm_example_training_mode",
-    "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
@@ -547,6 +546,7 @@
     "test_asin",
     "test_averagepool_2d_ceil",
     "test_bernoulli",
+    "test_bernoulli_double",
     "test_bernoulli_double_expanded",
     "test_bernoulli_expanded",
     "test_cast_BFLOAT16_to_FLOAT",
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
index 53955694ff7e..bac1369fd991 100644
--- a/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
+++ b/build_tools/pkgci/external_test_suite/onnx_gpu_rocm_rdna3.json
@@ -45,7 +45,6 @@
     "test_basic_deform_conv_without_padding",
     "test_batchnorm_epsilon_training_mode",
     "test_batchnorm_example_training_mode",
-    "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
@@ -542,6 +541,7 @@
     "test_asin",
     "test_averagepool_2d_ceil",
     "test_bernoulli",
+    "test_bernoulli_double",
     "test_bernoulli_double_expanded",
     "test_bernoulli_expanded",
     "test_cast_BFLOAT16_to_FLOAT",
diff --git a/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json b/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
index 91335de62e5f..21c894bb774e 100644
--- a/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
+++ b/build_tools/pkgci/external_test_suite/onnx_gpu_vulkan.json
@@ -60,7 +60,6 @@
     "test_basic_deform_conv_without_padding",
     "test_batchnorm_epsilon_training_mode",
     "test_batchnorm_example_training_mode",
-    "test_bernoulli_double",
     "test_bernoulli_seed",
     "test_bernoulli_seed_expanded",
     "test_cast_FLOAT16_to_FLOAT8E4M3FN",
@@ -587,6 +586,7 @@
     "test_asin",
     "test_averagepool_2d_ceil",
     "test_bernoulli",
+    "test_bernoulli_double",
     "test_bernoulli_double_expanded",
     "test_bernoulli_expanded",
     "test_bitshift_left_uint16",
diff --git a/third_party/torch-mlir b/third_party/torch-mlir
index 661be2d5b0ac..d59d0b6e5a88 160000
--- a/third_party/torch-mlir
+++ b/third_party/torch-mlir
@@ -1 +1 @@
-Subproject commit 661be2d5b0ac0936be4f9139b5b1be099905d885
+Subproject commit d59d0b6e5a88252d1d7e9b380e5488f49fadf87f