From 5a22d769072038f5e8e2e789fe8649c35e16546a Mon Sep 17 00:00:00 2001 From: Cody Yu Date: Fri, 25 Sep 2020 23:12:55 +0000 Subject: [PATCH] Make CMakefile/config.cmake/install_tvm consistent --- CMakeLists.txt | 4 ++-- docker/install/install_tvm_cpu.sh | 2 -- docker/install/install_tvm_gpu.sh | 3 --- 3 files changed, 2 insertions(+), 7 deletions(-) diff --git a/CMakeLists.txt b/CMakeLists.txt index c6906e832e39..a29364b4d779 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -65,9 +65,9 @@ tvm_option(USE_CUBLAS "Build with cuBLAS" OFF) tvm_option(USE_THRUST "Build with Thrust" OFF) tvm_option(USE_MIOPEN "Build with ROCM:MIOpen" OFF) tvm_option(USE_ROCBLAS "Build with ROCM:RoCBLAS" OFF) -tvm_option(USE_SORT "Build with sort support" OFF) +tvm_option(USE_SORT "Build with sort support" ON) tvm_option(USE_NNPACK "Build with nnpack support" OFF) -tvm_option(USE_RANDOM "Build with random support" OFF) +tvm_option(USE_RANDOM "Build with random support" ON) tvm_option(USE_MICRO_STANDALONE_RUNTIME "Build with micro.standalone_runtime support" OFF) tvm_option(USE_CPP_RPC "Build CPP RPC" OFF) tvm_option(USE_TFLITE "Build with tflite support" OFF) diff --git a/docker/install/install_tvm_cpu.sh b/docker/install/install_tvm_cpu.sh index 3dbf8e8e710f..b11c9791fb2d 100755 --- a/docker/install/install_tvm_cpu.sh +++ b/docker/install/install_tvm_cpu.sh @@ -27,8 +27,6 @@ cd /usr/tvm git checkout 4b13bf668edc7099b38d463e5db94ebc96c80470 echo set\(USE_LLVM llvm-config-8\) >> config.cmake -echo set\(USE_RPC ON\) >> config.cmake -echo set\(USE_SORT ON\) >> config.cmake echo set\(USE_GRAPH_RUNTIME ON\) >> config.cmake echo set\(USE_BLAS openblas\) >> config.cmake mkdir -p build diff --git a/docker/install/install_tvm_gpu.sh b/docker/install/install_tvm_gpu.sh index d9b9a0a183d1..2dbf8e17398d 100755 --- a/docker/install/install_tvm_gpu.sh +++ b/docker/install/install_tvm_gpu.sh @@ -29,9 +29,6 @@ git checkout 4b13bf668edc7099b38d463e5db94ebc96c80470 echo set\(USE_LLVM llvm-config-8\) >> config.cmake echo set\(USE_CUDA ON\) >> config.cmake echo set\(USE_CUDNN ON\) >> config.cmake -echo set\(USE_RPC ON\) >> config.cmake -echo set\(USE_SORT ON\) >> config.cmake -echo set\(USE_GRAPH_RUNTIME ON\) >> config.cmake echo set\(USE_BLAS openblas\) >> config.cmake mkdir -p build cd build