saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Run PadGrad UT for CUDA Only (#7947)

Browse source
This commit is contained in:
Vincent Wang 2021-06-04 22:32:08 +08:00 committed by GitHub
parent 650314c926
commit 2b3f953701
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/test/gradient/gradient_ops_test.cc
View file

@ -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