saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-03 23:49:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Disable GradientCheckerTest tests for GPU/Debug build (#3407)

Browse source
This commit is contained in:
Thiago Crepaldi 2020-04-02 18:00:54 -07:00
parent e2afe5e054
commit d89e5d91a6
2 changed files with 2 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cmake/onnxruntime_unittests.cmake
View file

@ -406,11 +406,6 @@ endif()
set(all_dependencies ${onnxruntime_test_providers_dependencies} )
if (onnxruntime_ENABLE_TRAINING)
if(${CMAKE_BUILD_TYPE} MATCHES "Debug" AND NOT onnxruntime_USE_CUDA)
file(GLOB grad_test_src "${ORTTRAINING_SOURCE_DIR}/test/gradient/*.cc")
list(REMOVE_ITEM onnxruntime_test_training_src ${grad_test_src})
endif()
list(APPEND all_tests ${onnxruntime_test_training_src})
endif()

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

@ -19,6 +19,7 @@
namespace onnxruntime {
namespace test {
#ifdef NDEBUG
using ONNX_NAMESPACE::MakeAttribute;
using training::OpDef;
@ -1563,6 +1564,7 @@ TEST(Synchronization, WaitAndRecordEventMany) {
}
}
}
#endif
} // namespace test
} // namespace onnxruntime