Update rust/tvm-rt/src/to_function.rs

Co-authored-by: Junru Shao <junrushao1994@gmail.com>
apache · Oct 24, 2020 · 55c6e9d · 55c6e9d
1 parent fb96d5c
commit 55c6e9d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/rust/tvm-rt/src/to_function.rs b/rust/tvm-rt/src/to_function.rs
@@ -103,10 +103,10 @@ pub trait ToFunction<I, O>: Sized {
                 value = args_list[i];
                 tcode = type_codes_list[i];
                 if tcode == ffi::TVMArgTypeCode_kTVMObjectHandle as c_int
-                    || tcode == ffi::TVMArgTypeCode_kkTVMObjectRValueRefArg as c_int
+                    || tcode == ffi::TVMArgTypeCode_kTVMObjectRValueRefArg as c_int
                     || tcode == ffi::TVMArgTypeCode_kTVMPackedFuncHandle as c_int
                     || tcode == ffi::TVMArgTypeCode_kTVMModuleHandle as c_int
-                    || tcode == ffi::TVMArgTypeCode_kTVMNDArrayFuncHandle as c_int
+                    || tcode == ffi::TVMArgTypeCode_kTVMNDArrayHandle as c_int
                 {
                     check_call!(ffi::TVMCbArgToReturn(
                         &mut value as *mut _,