You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by jw...@apache.org on 2021/04/09 19:17:28 UTC
[tvm] 07/09: Fix driver
This is an automated email from the ASF dual-hosted git repository.
jwfromm pushed a commit to branch checkpoint
in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 40d85a7d300976d48e9ec77e04dde4bc8bb99cd1
Author: Jared Roesch <ro...@gmail.com>
AuthorDate: Tue Mar 16 14:01:58 2021 -0700
Fix driver
---
src/driver/driver_api.cc | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/driver/driver_api.cc b/src/driver/driver_api.cc
index 9172283..0011dab 100644
--- a/src/driver/driver_api.cc
+++ b/src/driver/driver_api.cc
@@ -356,7 +356,7 @@ void TVMExtractOutputShapes(tvm::runtime::Module& mod, size_t num_outputs, std::
}
}
-void TVMRun(tvm::runtime::Module& mod, std::vector<DLTensor>& inputs, std::vector<DLTensor>& outputs)
+void TVMRun(tvm::runtime::Module& mod, std::vector<DLTensor>& inputs, std::vector<DLTensor>& outputs, tvm::runtime::TVMRetValue *ret)
{
tvm::PackedFunc set_input = mod.GetFunction("set_input_zero_copy", false);
for (size_t i = 0; i < inputs.size(); i++)