From 7456de417d045c3c72749a11d897c412a4096818 Mon Sep 17 00:00:00 2001
From: Prathik Rao
 <prathikrao@microsoft.com@orttrainingdev8.d32nl1ml4oruzj4qz3bqlggovf.px.internal.cloudapp.net>
Date: Wed, 24 Jan 2024 00:04:38 +0000
Subject: [PATCH] outputs to 3

---
 orttraining/orttraining/test/gradient/gradient_ops_test.cc | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/orttraining/orttraining/test/gradient/gradient_ops_test.cc b/orttraining/orttraining/test/gradient/gradient_ops_test.cc
index 95e5a9194efb8..e007a0791e843 100644
--- a/orttraining/orttraining/test/gradient/gradient_ops_test.cc
+++ b/orttraining/orttraining/test/gradient/gradient_ops_test.cc
@@ -700,7 +700,7 @@ TEST(GradientCheckerTest, SplitGrad) {
 
   ASSERT_STATUS_OK(gradient_checker.ComputeGradientError(op_def, {shape}, {{3, 5}, {3, 5}, {3, 5}}, &max_error,
                                                          {MakeAttribute("axis", int64_t(0)),
-                                                          MakeAttribute("num_outputs", int64_t(2))}));
+                                                          MakeAttribute("num_outputs", int64_t(3))}));
   EXPECT_IS_TINY(max_error);
 
   // opset13 test