This is an automated email from the ASF dual-hosted git repository. jroesch pushed a commit to branch tvm_ep in repository https://gitbox.apache.org/repos/asf/tvm.git
commit dbcf2e47eff4b818c08f972e5d351acdc4394615 Author: Jared Roesch <[email protected]> AuthorDate: Fri Mar 5 14:02:38 2021 -0800 Fix --- include/tvm/driver/jit_interface.h | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/include/tvm/driver/jit_interface.h b/include/tvm/driver/jit_interface.h index 47b98c9..c4596e9 100644 --- a/include/tvm/driver/jit_interface.h +++ b/include/tvm/driver/jit_interface.h @@ -4,9 +4,6 @@ extern "C" { EXPORT_DLL tvm::runtime::Module TVMCompile(const std::string& onnx_txt, const std::string& target, const std::string& target_host, int opt_level, const std::vector<std::vector<int64_t>>& input_shapes); EXPORT_DLL void TVMExtractOutputShapes(tvm::runtime::Module& mod, size_t num_outputs, std::vector<std::vector<int64_t>>& output_shapes); - - EXPORT_DLL void TVMRun(tvm::runtime::Module& mod, std::vector<DLTensor>& inputs, std::vector<DLTensor>& outputs); - - + EXPORT_DLL void TVMRun(tvm::runtime::Module& mod, std::vector<DLTensor>& inputs, std::vector<DLTensor>& outputs, tvm::runtime::TVMRetValue *ret); } // TVM_EXTERN_C #endif
