40 lines
1.6 KiB
CMake
40 lines
1.6 KiB
CMake
#===============================================================================
|
|
# Copyright 2018 Intel Corporation
|
|
#
|
|
# Licensed 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.
|
|
#===============================================================================
|
|
|
|
# Utils for managing threading-related configuration
|
|
#===============================================================================
|
|
|
|
if(Threading_cmake_included)
|
|
return()
|
|
endif()
|
|
set(Threading_cmake_included true)
|
|
|
|
# Replace existing define for threading (if any) with a new one
|
|
macro(set_threading threading)
|
|
if(MKLDNN_THR_CURRENT)
|
|
remove_definitions(-DMKLDNN_THR=${MKLDNN_THR_CURRENT})
|
|
endif()
|
|
set(MKLDNN_THR_CURRENT MKLDNN_THR_${threading})
|
|
add_definitions(-DMKLDNN_THR=${MKLDNN_THR_CURRENT})
|
|
endmacro()
|
|
|
|
# While MKL-DNN defaults to OpenMP (if _OPENMP is defined) without CMake, here
|
|
# we default to sequential threading and let OpenMP.cmake and TBB.cmake to
|
|
# figure things out. This is especially important because OpenMP is used both
|
|
# for threading and vectorization via #pragma omp simd
|
|
set_threading("SEQ")
|
|
|