From 08f512b25e1624be166d6dd0284e2e737b8c994a Mon Sep 17 00:00:00 2001 From: Ye Wang <52801275+wangyems@users.noreply.github.com> Date: Thu, 6 Jan 2022 09:46:34 -0800 Subject: [PATCH] Fix a Win GPU reduced ops pipeline (#10202) --- onnxruntime/test/shared_lib/test_inference.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/onnxruntime/test/shared_lib/test_inference.cc b/onnxruntime/test/shared_lib/test_inference.cc index 00a42f1af1..4d033432d8 100644 --- a/onnxruntime/test/shared_lib/test_inference.cc +++ b/onnxruntime/test/shared_lib/test_inference.cc @@ -744,6 +744,9 @@ TEST(CApiTest, RegisterCustomOpForCPUAndCUDA) { //It has memory leak. The OrtCustomOpDomain created in custom_op_library.cc:RegisterCustomOps function was not freed #if defined(__ANDROID__) TEST(CApiTest, DISABLED_test_custom_op_library) { +//To accomodate a reduced op build pipeline +#elif defined(REDUCED_OPS_BUILD) && defined(USE_CUDA) +TEST(CApiTest, DISABLED_test_custom_op_library) { #else TEST(CApiTest, test_custom_op_library) { #endif