threadpool.cmake 1.5 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
# Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserved.
#
# 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.

Y
Yu Yang 已提交
15 16
INCLUDE(ExternalProject)

17
SET(THREADPOOL_PREFIX_DIR ${THIRD_PARTY_PATH}/threadpool)
18
if(WITH_ASCEND OR WITH_ASCEND_CL)
19 20 21 22
    SET(THREADPOOL_REPOSITORY https://gitee.com/tianjianhe/ThreadPool.git)
else()
    SET(THREADPOOL_REPOSITORY ${GIT_URL}/progschj/ThreadPool.git)
endif()
23
SET(THREADPOOL_TAG        9a42ec1329f259a5f4881a291db1dcb8f2ad9040)
Y
Yu Yang 已提交
24

25
SET(THREADPOOL_INCLUDE_DIR ${THIRD_PARTY_PATH}/threadpool/src/extern_threadpool)
26
INCLUDE_DIRECTORIES(${THREADPOOL_INCLUDE_DIR})
27

Y
Yu Yang 已提交
28 29 30
ExternalProject_Add(
    extern_threadpool
    ${EXTERNAL_PROJECT_LOG_ARGS}
31
    ${SHALLOW_CLONE}
32 33
    GIT_REPOSITORY  ${THREADPOOL_REPOSITORY}
    GIT_TAG         ${THREADPOOL_TAG}
34
    PREFIX          ${THREADPOOL_PREFIX_DIR}
Y
Yu Yang 已提交
35 36 37 38 39 40 41
    UPDATE_COMMAND  ""
    CONFIGURE_COMMAND ""
    BUILD_COMMAND     ""
    INSTALL_COMMAND   ""
    TEST_COMMAND      ""
)

42
add_library(simple_threadpool INTERFACE)
Y
Yu Yang 已提交
43 44

add_dependencies(simple_threadpool extern_threadpool)