diff --git a/CMakeLists.txt b/CMakeLists.txt index b43d37696b37..4ec5ea779f96 100644 --- 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 index 1356c8560f3b..43fb700203c7 100644 --- 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..")