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 a80e07a2950df..d9800ce0e0d3e 100644 --- a/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc +++ b/orttraining/orttraining/test/training_ops/cuda/cross_entropy_test.cc @@ -641,7 +641,7 @@ TEST(CrossEntropyTest, DISABLED_SoftmaxCrossEntropyLoss_LargeSizeTensor) { #ifndef _WIN32 // Disable the large size tests for Windows because it is too slow, running on Linux would be enough. // This test requires lots of memory, currently, it can run with 16GB V100 GPU. -TEST(CrossEntropyTest, SoftmaxCrossEntropyLossInternal_LargeSizeTensorUInt64Index) { +TEST(CrossEntropyTest, DISABLED_SoftmaxCrossEntropyLossInternal_LargeSizeTensorUInt64Index) { // The element count is bigger than the upper limit of int32_t. constexpr int64_t bsz = 419431; constexpr int64_t vocab_size = 5120; @@ -1073,7 +1073,7 @@ TEST(CrossEntropyTest, SoftmaxCrossEntropyLossInternalGrad_TinySizeTensorFloatIn #ifndef _WIN32 // Disable the large size tests for Windows because it is too slow, running on Linux would be enough. // This test requires lots of memory, currently, it can run with 16GB V100 GPU. -TEST(CrossEntropyTest, SoftmaxCrossEntropyLossInternalGrad_LargeSizeTensorUInt64Index) { +TEST(CrossEntropyTest, DISABLED_SoftmaxCrossEntropyLossInternalGrad_LargeSizeTensorUInt64Index) { // The element count is bigger than the upper limit of int32_t. constexpr int64_t bsz = 419431; constexpr int64_t vocab_size = 5120;