From 2b3f953701310a4f604d6f93250e9ab4434a0802 Mon Sep 17 00:00:00 2001
From: Vincent Wang <wangwchpku@outlook.com>
Date: Fri, 4 Jun 2021 22:32:08 +0800
Subject: [PATCH] Run PadGrad UT for CUDA Only (#7947)

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

diff --git a/orttraining/orttraining/test/gradient/gradient_ops_test.cc b/orttraining/orttraining/test/gradient/gradient_ops_test.cc
index 782805c975..c199592637 100755
--- a/orttraining/orttraining/test/gradient/gradient_ops_test.cc
+++ b/orttraining/orttraining/test/gradient/gradient_ops_test.cc
@@ -2580,6 +2580,7 @@ TEST(GradientCheckerTest, TileGrad) {
   }
 }
 
+#ifdef USE_CUDA
 TEST(GradientCheckerTest, PadGrad) {
   float max_error;
   GradientChecker<float, float, float> gradient_checker;
@@ -2648,6 +2649,7 @@ TEST(GradientCheckerTest, PadGrad) {
     ASSERT_TRUE(has_error);
   }
 }
+#endif  // USE_CUDA
 
 }  // namespace test
 }  // namespace onnxruntime