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:27 UTC
[tvm] 06/09: Fix
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 dbcf2e47eff4b818c08f972e5d351acdc4394615
Author: Jared Roesch <jr...@octoml.ai>
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