From 2e4b9b1d0e332abdddffe57df4d9f99ae6151bf3 Mon Sep 17 00:00:00 2001
From: Edward Chen <18449977+edgchen1@users.noreply.github.com>
Date: Tue, 21 Apr 2020 20:30:45 +0000
Subject: [PATCH] Disable
 CudaKernelTest.SoftmaxCrossEntropyLoss_LargeSizeTensor because it's flaky.

---
 .../orttraining/test/training_ops/cuda/cross_entropy_test.cc  | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc b/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc
index 638108b98d..1a3dcda87f 100644
--- a/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc
+++ b/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc
@@ -363,7 +363,9 @@ TEST(CudaKernelTest, SoftmaxCrossEntropyLoss_MediumSizeTensor) {
   TestSoftmaxCrossEntropyLoss(&X_dims, &index_dims, nullptr, &Y_dims_none, &log_prob_dims, "none");
 }
 
-TEST(CudaKernelTest, SoftmaxCrossEntropyLoss_LargeSizeTensor) {
+// TODO fix flaky test (https://msdata.visualstudio.com/Vienna/_workitems/edit/596949)
+// failing random seed: 2873512643
+TEST(CudaKernelTest, DISABLED_SoftmaxCrossEntropyLoss_LargeSizeTensor) {
   std::vector<int64_t> X_dims{4, 512, 30528};
   std::vector<int64_t> index_dims{4, 30528};
   std::vector<int64_t> weight_dims{512};