fix: TVM EP register function definition for non-load-dynamic, closes

#227
pykeio · Jul 6, 2024 · 1bff72d · 1bff72d
1 parent bf10c18
commit 1bff72d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/execution_providers/tvm.rs b/src/execution_providers/tvm.rs
@@ -6,7 +6,7 @@ use crate::{
 
 #[cfg(all(not(feature = "load-dynamic"), feature = "tvm"))]
 extern "C" {
-	fn OrtSessionOptionsAppendExecutionProvider_Tvm(options: *mut ort_sys::OrtSessionOptions, use_arena: std::os::raw::c_int) -> ort_sys::OrtStatusPtr;
+	fn OrtSessionOptionsAppendExecutionProvider_Tvm(options: *mut ort_sys::OrtSessionOptions, opt_str: *const std::os::raw::c_char) -> ort_sys::OrtStatusPtr;
 }
 
 #[derive(Debug, Clone, Copy, PartialEq, Eq)]