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

adityagoel4512 · Jun 30, 2023 · 5b3447b · 5b3447b
1 parent 0051497
commit 5b3447b
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op.cc b/onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op.cc
@@ -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
   }
 }
 

diff --git a/onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op_lib.cc b/onnxruntime/test/testdata/custom_op_get_const_input_test_library/custom_op_lib.cc
@@ -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;
 }