diff --git a/CMakeLists.txt b/CMakeLists.txt index ff6a5f69e28c1..2c53eb3cf7f0b 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -57,6 +57,7 @@ tvm_option(PICOJSON_PATH "Path to PicoJSON" "3rdparty/picojson") tvm_option(USE_BLAS "The blas library to be linked" none) tvm_option(USE_MKL_PATH "MKL root path when use MKL blas" none) tvm_option(USE_MKLDNN "Build with MKLDNN" OFF) +tvm_option(USE_DNNL_CODEGEN "Enable MKLDNN (DNNL) codegen" OFF) tvm_option(USE_CUDNN "Build with cuDNN" OFF) tvm_option(USE_CUBLAS "Build with cuBLAS" OFF) tvm_option(USE_THRUST "Build with Thrust" OFF)