[BUILD] enhance vta build (apache#1434)

tqchen · Jul 15, 2018 · 6b8d0c0 · 6b8d0c0
1 parent 12cf343
commit 6b8d0c0
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/CMakeLists.txt b/CMakeLists.txt
@@ -35,8 +35,6 @@ tvm_option(USE_RTTI "Build with RTTI" ON)
 tvm_option(USE_MSVC_MT "Build with MT" OFF)
 tvm_option(INSTALL_DEV "Install compiler infrastructure" OFF)
 
-tvm_option(USE_VTA_CFG "Use a specific json file for VTA runtime" "")
-
 # Contrib library options
 tvm_option(USE_BLAS "The blas library to be linked" none)
 tvm_option(USE_MKL_PATH "MKL root path when use MKL blas" none)

diff --git a/cmake/modules/VTA.cmake b/cmake/modules/VTA.cmake
@@ -1,5 +1,5 @@
 # CMake Build rules for VTA
-find_program(PYTHON python)
+find_program(PYTHON NAMES python python3 python3.6)
 
 if(MSVC)
   message(STATUS "VTA build is skipped in Windows..")