[Rust] fix #6205 (#6207)

diff --git a/rust/tvm-sys/src/byte_array.rs b/rust/tvm-sys/src/byte_array.rs
index ea2feeb..0f02771 100644 (file)
--- a/rust/tvm-sys/src/byte_array.rs
+++ b/rust/tvm-sys/src/byte_array.rs
@@ -72,10 +72,9 @@ impl<T: AsRef<[u8]>> From<T> for ByteArray {
     }
 }
 
-impl From<ByteArray> for ArgValue<'static> {
-    fn from(val: ByteArray) -> ArgValue<'static> {
-        // TODO(@jroesch): brorowed ArgValue are not sound
-        ArgValue::Bytes(unsafe { std::mem::transmute(&val.array) })
+impl<'a> From<&'a ByteArray> for ArgValue<'a> {
+    fn from(val: &'a ByteArray) -> ArgValue<'a> {
+        ArgValue::Bytes(&val.array)
     }
 }
 

diff --git a/rust/tvm-sys/src/packed_func.rs b/rust/tvm-sys/src/packed_func.rs
index b1e2af9..3121deb 100644 (file)
--- a/rust/tvm-sys/src/packed_func.rs
+++ b/rust/tvm-sys/src/packed_func.rs
@@ -161,7 +161,7 @@ TVMPODValue! {
     },
     match &self {
         Bytes(val) => {
-            (TVMValue { v_handle: val as *const _ as *mut c_void }, TVMArgTypeCode_kTVMBytes)
+            (TVMValue { v_handle: *val as *const _ as *mut c_void }, TVMArgTypeCode_kTVMBytes)
         }
         Str(val) => { (TVMValue { v_handle: val.as_ptr() as *mut c_void }, TVMArgTypeCode_kTVMStr) }
     }

diff --git a/rust/tvm/examples/resnet/src/main.rs b/rust/tvm/examples/resnet/src/main.rs
index c81087d..16ca8c7 100644 (file)
--- a/rust/tvm/examples/resnet/src/main.rs
+++ b/rust/tvm/examples/resnet/src/main.rs
@@ -91,7 +91,7 @@ fn main() {
         fs::read(concat!(env!("CARGO_MANIFEST_DIR"), "/deploy_param.params")).unwrap();
     let barr = ByteArray::from(&params);
     // load the parameters
-    load_param_fn.invoke(vec![barr.into()]).unwrap();
+    load_param_fn.invoke(vec![(&barr).into()]).unwrap();
     // get the set_input function
     let ref set_input_fn = graph_runtime_module
         .get_function("set_input", false)