From 1bff72d9780490ff7856528e6bc3997528fd1714 Mon Sep 17 00:00:00 2001 From: "Carson M." Date: Sat, 6 Jul 2024 09:11:22 -0500 Subject: [PATCH] fix: TVM EP register function definition for non-`load-dynamic`, closes #227 --- src/execution_providers/tvm.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/execution_providers/tvm.rs b/src/execution_providers/tvm.rs index 6a04d94..6e43601 100644 --- 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)]