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

Allow disable exceptions to work with ort-extensions (#16536)

Browse source
This commit is contained in:
Baiju Meswani 2023-06-29 18:24:33 -07:00 committed by GitHub
parent 0051497055
commit 5b3447beef
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op.cc vendored
View file

@ -5,7 +5,12 @@
void simple_assert(const bool cond, const std::string& text) {
if (!cond) {
#ifndef ORT_NO_EXCEPTIONS
throw std::runtime_error(text);
#else
std::cerr << text << std::endl;
std::terminate();
#endif
}
}

4
onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op_lib.cc vendored
View file

@ -25,17 +25,21 @@ OrtStatus* ORT_API_CALL RegisterCustomOps(OrtSessionOptions* options, const OrtA
OrtStatus* result = nullptr;
#ifndef ORT_NO_EXCEPTIONS
try {
#endif
Ort::CustomOpDomain domain{c_OpDomain};
domain.Add(&c_CustomOp);
session_options.Add(domain);
AddOrtCustomOpDomainToContainer(std::move(domain));
#ifndef ORT_NO_EXCEPTIONS
} catch (const std::exception& e) {
Ort::Status status{e};
result = status.release();
}
#endif
return result;
}