From 5ffd58c8e6574934a8272c7547697db32a9c8828 Mon Sep 17 00:00:00 2001 From: Hariharan Seshadri Date: Thu, 6 Jul 2023 14:32:59 -0700 Subject: [PATCH] Fix Reduced Ops pipeline (#16612) --- onnxruntime/test/shared_lib/test_inference.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/onnxruntime/test/shared_lib/test_inference.cc b/onnxruntime/test/shared_lib/test_inference.cc index cc27d78632..e4640187a8 100644 --- a/onnxruntime/test/shared_lib/test_inference.cc +++ b/onnxruntime/test/shared_lib/test_inference.cc @@ -1887,6 +1887,8 @@ TEST(CApiTest, basic_cuda_graph) { binding.ClearBoundOutputs(); } +#ifndef REDUCED_OPS_BUILD +// The following test uses some ops not supported in the reduced ops build TEST(CApiTest, cuda_graph_with_shape_nodes) { const auto& api = Ort::GetApi(); @@ -1910,6 +1912,8 @@ TEST(CApiTest, cuda_graph_with_shape_nodes) { #endif +#endif + TEST(CApiTest, create_tensor) { const char* s[] = {"abc", "kmp"}; int64_t expected_len = 2;