saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-12 17:57:38 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix Reduced Ops pipeline (#16612)

Browse source
This commit is contained in:
Hariharan Seshadri 2023-07-06 14:32:59 -07:00 committed by GitHub
parent 51c42ae64a
commit 5ffd58c8e6
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
onnxruntime/test/shared_lib/test_inference.cc
View file

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