diff --git a/CMakeLists.txt b/CMakeLists.txt index 1d11d2f0ee858949210c73da5c90dfa16c80d5e5..25d45c0b28cf12a1eb222a0c50d6b1c5849e3bd7 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -776,6 +776,7 @@ if (UNIX OR APPLE) SOURCE_DIR ${CAFFE_URL} PREFIX ${CAFFE_PREFIX} CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH= + -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TOOLCHAIN_FILE} -DMKLDNN_INSTALL_DIR:PATH= -DUSE_MKL2017_AS_DEFAULT_ENGINE=${CAFFE_CPU_ONLY} -DUSE_CUDNN=${USE_CUDNN} @@ -796,6 +797,7 @@ if (UNIX OR APPLE) SOURCE_DIR ${CAFFE_URL} PREFIX ${CAFFE_PREFIX} CMAKE_ARGS -DCMAKE_INSTALL_PREFIX:PATH= + -DCMAKE_TOOLCHAIN_FILE=${CMAKE_TOOLCHAIN_FILE} -DUSE_CUDNN=${USE_CUDNN} -DCUDA_ARCH_NAME=${CUDA_ARCH} -DCUDA_ARCH_BIN=${CUDA_ARCH_BIN}