diff --git a/orttraining/orttraining/test/training_ops/cuda/negativeloglikelihood_test.cc b/orttraining/orttraining/test/training_ops/cuda/negativeloglikelihood_test.cc index 184be72fcb..2823655461 100644 --- a/orttraining/orttraining/test/training_ops/cuda/negativeloglikelihood_test.cc +++ b/orttraining/orttraining/test/training_ops/cuda/negativeloglikelihood_test.cc @@ -92,7 +92,7 @@ TEST(CudaKernelTest, NegativeLogLikelihoodLoss_MediumSizeTensor) { TestNegativeLogLikelihoodLoss(&X_dims, &index_dims, nullptr, &Y_dims_none, "none"); } -TEST(CudaKernelTest, NegativeLogLikelihoodLoss_LargeSizeTensor) { +TEST(CudaKernelTest, DISABLED_NegativeLogLikelihoodLoss_LargeSizeTensor) { std::vector X_dims{4, 512, 30528}; std::vector index_dims{4, 30528}; std::vector weight_dims{512};