From 6d9d252bc3aa9821829c3dfccac5c6e33e3b363d Mon Sep 17 00:00:00 2001 From: "M. Zeeshan Siddiqui" Date: Mon, 31 Aug 2020 11:02:07 -0700 Subject: [PATCH] Disable NegativeLogLikelihoodLoss_LargeSizeTensor test (#4979) Disabling this test until it's intermittent failure is root caused, this is a function and does not have a dedicated op by itself. However, this op is not used in known model to the best of my knowledge to disabling this test for the sanity of CI until the investigation is over is probably reasonable. --- .../test/training_ops/cuda/negativeloglikelihood_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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};