lebeg commented on a change in pull request #11148: [MXNET-679] Refactor handling BLAS libraries with cmake URL: https://github.com/apache/incubator-mxnet/pull/11148#discussion_r209664120
########## File path: cmake/ChooseBLAS.cmake ########## @@ -0,0 +1,266 @@ +# Licensed to the Apache Software Foundation (ASF) under one +# or more contributor license agreements. See the NOTICE file +# distributed with this work for additional information +# regarding copyright ownership. The ASF licenses this file +# to you under the Apache License, Version 2.0 (the +# "License"); you may not use this file except in compliance +# with the License. You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, +# software distributed under the License is distributed on an +# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY +# KIND, either express or implied. See the License for the +# specific language governing permissions and limitations +# under the License. + +set(BLAS "Open" CACHE STRING "Selected BLAS library") +set_property(CACHE BLAS PROPERTY STRINGS "Atlas;Open;MKL;Apple") + +function(switch_lapack ENABLE) + if(ENABLE) + message(STATUS "Enabling LAPACK functionality") + add_definitions(-DMXNET_USE_LAPACK=1) + else() + if(USE_LAPACK) + message(WARNING "LAPACK functionality not available") + else() + message(STATUS "LAPACK functionality not available") + endif() + endif() +endfunction() + +function(try_mkldnn) + + if(NOT USE_MKLDNN) + return() + endif() + + message(STATUS "Adding MKLDNN to the build due to \ + USE_MKLDNN=${USE_MKLDNN} and USE_MKL_IF_AVAILABLE=${USE_MKL_IF_AVAILABLE}") + + # CPU architecture (e.g., C5) can't run on another architecture (e.g., g3). + if(NOT MSVC) + set(ARCH_OPT_FLAGS ${ARCH_OPT_FLAGS} "-mtune=generic" PARENT_SCOPE) + endif() + + if(MSVC) + file(COPY ${CMAKE_SOURCE_DIR}/3rdparty/mkldnn/config_template.vcxproj.user DESTINATION ${CMAKE_SOURCE_DIR}) + endif() + + set(WITH_TEST OFF) + set(WITH_EXAMPLE OFF) + add_subdirectory(3rdparty/mkldnn) + + include_directories(3rdparty/mkldnn/include) + set(mxnet_LINKER_LIBS ${mxnet_LINKER_LIBS} mkldnn PARENT_SCOPE) + + add_definitions(-DMXNET_USE_MKLDNN=1) + +endfunction() + +function(try_mkl) + + message(STATUS "Trying to enable MKL framework due to USE_MKL_IF_AVAILABLE=${USE_MKL_IF_AVAILABLE}") + + if(CMAKE_CROSSCOMPILING) + message(WARNING "MKL with cross compilation is not supported, MKL will not be available") + return() + endif() + + if(NOT SYSTEM_ARCHITECTURE STREQUAL "x86_64") + message(WARNING "MKL is supported only for desktop platforms (SYSTEM_ARCHITECTURE=${SYSTEM_ARCHITECTURE}), \ + MKL will not be available") + return() + endif() + + find_package(MKL) + + if(MKL_FOUND) + message(STATUS "MKL framework found") + + set(MKL_FOUND ${MKL_FOUND} PARENT_SCOPE) + set(MKL_INCLUDE_DIR ${MKL_INCLUDE_DIR} PARENT_SCOPE) + set(MKL_LIBRARIES ${MKL_LIBRARIES} PARENT_SCOPE) + set(MKLROOT ${MKLROOT} PARENT_SCOPE) + + set(BLAS MKL PARENT_SCOPE) + else() + message(STATUS "MKL framework not found") + endif() + +endfunction() + +function(try_mklml) + if(NOT USE_MKLML) + return() + endif() + + if(MKL_FOUND) + return() + endif() + + message(STATUS "Trying to enable MKLML framework due to \ + USE_MKLML=${USE_MKLML} and USE_MKL_IF_AVAILABLE=${USE_MKL_IF_AVAILABLE}") + + if(CMAKE_CROSSCOMPILING) + message(WARNING "MKLML with cross compilation is not supported, MKL will not be available") + return() + endif() + + if(NOT SYSTEM_ARCHITECTURE STREQUAL "x86_64") + message(WARNING "MKL is supported only for desktop platforms (SYSTEM_ARCHITECTURE=${SYSTEM_ARCHITECTURE}), \ + MKL will not be available") + return() + endif() + + include(${CMAKE_CURRENT_LIST_DIR}/DownloadMKLML.cmake) + find_package(MKLML REQUIRED) + + set(MKL_FOUND ${MKL_FOUND} PARENT_SCOPE) + set(MKL_INCLUDE_DIR ${MKL_INCLUDE_DIR} PARENT_SCOPE) + set(MKL_LIBRARIES ${MKL_LIBRARIES} PARENT_SCOPE) + set(MKLROOT ${MKLROOT} PARENT_SCOPE) + + set(BLAS MKL PARENT_SCOPE) + + message(STATUS "MKLML framework found") + +endfunction() + +function(try_accelerate) + if(NOT APPLE) + return() + endif() + + if(BLAS MATCHES "[Mm][Kk][Ll]") + return() + endif() + + if(USE_APPLE_ACCELERATE_IF_AVAILABLE) + message(STATUS "Trying to enable Apple Accelerate framework due to USE_ACCELERATE_IF_AVAILABLE") + find_package(Accelerate) + if(Accelerate_FOUND) + message(STATUS "Apple Accelerate framework found") + set(BLAS Accelerate PARENT_SCOPE) + else() + message(STATUS "Apple Accelerate framework not found") + endif() + endif() +endfunction() + +if(USE_MKL_IF_AVAILABLE) + set(MKL_FOUND) + + try_mkl() + try_mklml() + try_mkldnn() +endif() + +try_accelerate() + +# cmake regexp does not support case insensitive match (?i) or //i +if(BLAS MATCHES "[Aa][Tt][Ll][Aa][Ss]") + message(STATUS "Using Atlas for BLAS") + + set(Atlas_NEED_LAPACK ${USE_LAPACK}) + find_package(Atlas REQUIRED) + + include_directories(SYSTEM ${Atlas_INCLUDE_DIRS}) + set(mxnet_LINKER_LIBS ${mxnet_LINKER_LIBS} ${Atlas_LIBRARIES}) + + add_definitions(-DMSHADOW_USE_CBLAS=1) + add_definitions(-DMSHADOW_USE_MKL=0) + + if(USE_LAPACK AND Atlas_LAPACK_FOUND) + switch_lapack(True) + else() + switch_lapack(False) + endif() + + return() +endif() + +if(BLAS MATCHES "[Oo][Pp][Ee][Nn]") Review comment: Done ---------------------------------------------------------------- This is an automated message from the Apache Git Service. To respond to the message, please log on GitHub and use the URL above to go to the specific comment. For queries about this service, please contact Infrastructure at: us...@infra.apache.org With regards, Apache Git Services
