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 <roesch...@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++)