diff --git a/CMakeLists.txt b/CMakeLists.txt index 2e5f5724cde..0f28b2e15b5 100644 --- a/CMakeLists.txt +++ b/CMakeLists.txt @@ -204,6 +204,7 @@ option(USE_GLOG "Use GLOG" OFF) option(USE_LEVELDB "Use LEVELDB" OFF) option(USE_LITE_PROTO "Use lite protobuf instead of full." OFF) option(USE_LMDB "Use LMDB" OFF) +option(USE_MAGMA "Use MAGMA" ON) option(USE_METAL "Use Metal for Caffe2 iOS build" ON) option(USE_PYTORCH_METAL "Use Metal for PyTorch iOS build" OFF) option(USE_NATIVE_ARCH "Use -march=native" OFF) diff --git a/cmake/Dependencies.cmake b/cmake/Dependencies.cmake index 81f945bc77d..01558ae162c 100644 --- a/cmake/Dependencies.cmake +++ b/cmake/Dependencies.cmake @@ -1578,7 +1578,9 @@ if(NOT INTERN_BUILD_MOBILE) set(CUDA_ATTACH_VS_BUILD_RULE_TO_CUDA_FILE OFF) - find_package(MAGMA) + if(USE_MAGMA) + find_package(MAGMA) + endif() if((USE_CUDA OR USE_ROCM) AND MAGMA_FOUND) include_directories(SYSTEM ${MAGMA_INCLUDE_DIR}) if(USE_CUDA) @@ -1602,8 +1604,14 @@ if(NOT INTERN_BUILD_MOBILE) message(STATUS "MAGMA INCLUDE DIRECTORIES: ${MAGMA_INCLUDE_DIR}") message(STATUS "MAGMA LIBRARIES: ${MAGMA_LIBRARIES}") message(STATUS "MAGMA V2 check: ${MAGMA_V2}") + elseif(USE_MAGMA) + message(WARNING + "Not compiling with MAGMA. Suppress this warning with " + "-DUSE_MAGMA=OFF.") + caffe2_update_option(USE_MAGMA OFF) else() message(STATUS "MAGMA not found. Compiling without MAGMA support") + caffe2_update_option(USE_MAGMA OFF) endif() # ARM specific flags