From 2465cdaf287c41af0254d2cc7a4ea7a9aa6e2462 Mon Sep 17 00:00:00 2001 From: reminisce Date: Mon, 11 Nov 2019 00:40:31 -0800 Subject: [PATCH 1/2] Update tvm --- .gitmodules | 2 +- 3rdparty/tvm | 2 +- Makefile | 2 +- cmake/BuildTVM.cmake | 3 +++ 4 files changed, 6 insertions(+), 3 deletions(-) diff --git a/.gitmodules b/.gitmodules index 90ef157f0eec..1900820d4c86 100644 --- a/.gitmodules +++ b/.gitmodules @@ -19,7 +19,7 @@ branch = master [submodule "3rdparty/tvm"] path = 3rdparty/tvm - url = https://github.com/dmlc/tvm + url = https://github.com/apache/incubator-tvm.git [submodule "3rdparty/onnx-tensorrt"] path = 3rdparty/onnx-tensorrt url = https://github.com/onnx/onnx-tensorrt.git diff --git a/3rdparty/tvm b/3rdparty/tvm index afd4b3e44509..cff62bdbd20b 160000 --- a/3rdparty/tvm +++ b/3rdparty/tvm @@ -1 +1 @@ -Subproject commit afd4b3e4450984358e9d79a7e8e578483cb7b017 +Subproject commit cff62bdbd20bab3ebcc27213be1e6d79da00c842 diff --git a/Makefile b/Makefile index 841ba7e322ca..925cc3528375 100644 --- a/Makefile +++ b/Makefile @@ -612,7 +612,7 @@ lib/libtvm_runtime.so: [ -e $(LLVM_PATH)/bin/llvm-config ] || sh $(ROOTDIR)/contrib/tvmop/prepare_tvm.sh; \ cd $(TVM_PATH)/build; \ cmake -DUSE_LLVM="$(LLVM_PATH)/bin/llvm-config" \ - -DUSE_SORT=OFF -DUSE_CUDA=$(TVM_USE_CUDA) -DUSE_CUDNN=OFF ..; \ + -DUSE_SORT=OFF -DUSE_CUDA=$(TVM_USE_CUDA) -DUSE_CUDNN=OFF -DUSE_OPENMP=ON ..; \ $(MAKE) VERBOSE=1; \ mkdir -p $(ROOTDIR)/lib; \ cp $(TVM_PATH)/build/libtvm_runtime.so $(ROOTDIR)/lib/libtvm_runtime.so; \ diff --git a/cmake/BuildTVM.cmake b/cmake/BuildTVM.cmake index db8b33b84596..7fcf706ffb7c 100644 --- a/cmake/BuildTVM.cmake +++ b/cmake/BuildTVM.cmake @@ -133,3 +133,6 @@ set(USE_VTA_TSIM OFF) # Whether use Relay debug mode set(USE_RELAY_DEBUG OFF) + +# Use OPENMP thread pool to be compatible with MXNet +set(USE_OPENMP ON) From d57046eff8ecbd58c4672a03de9e6d9ec6a4417b Mon Sep 17 00:00:00 2001 From: reminisce Date: Thu, 14 Nov 2019 12:28:38 -0800 Subject: [PATCH 2/2] update tvm for cuda with fp16 --- 3rdparty/tvm | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/3rdparty/tvm b/3rdparty/tvm index cff62bdbd20b..5d66e7a695d1 160000 --- a/3rdparty/tvm +++ b/3rdparty/tvm @@ -1 +1 @@ -Subproject commit cff62bdbd20bab3ebcc27213be1e6d79da00c842 +Subproject commit 5d66e7a695d1d40e08cf2da802b5e2b7015b758a