Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into refine_layer_norm
sneaxiy
7 years ago
commit
010883689c
@ -0,0 +1,35 @@
|
||||
if(NOT WITH_GPU)
|
||||
return()
|
||||
endif()
|
||||
|
||||
include(ExternalProject)
|
||||
|
||||
set(CUB_SOURCE_DIR ${THIRD_PARTY_PATH}/cub)
|
||||
set(CUB_INCLUDE_DIR ${CUB_SOURCE_DIR}/src/extern_cub)
|
||||
|
||||
include_directories(${CUB_INCLUDE_DIR})
|
||||
|
||||
ExternalProject_Add(
|
||||
extern_cub
|
||||
${EXTERNAL_PROJECT_LOG_ARGS}
|
||||
GIT_REPOSITORY "https://github.com/NVlabs/cub.git"
|
||||
GIT_TAG "v1.8.0"
|
||||
PREFIX ${CUB_SOURCE_DIR}
|
||||
UPDATE_COMMAND ""
|
||||
CONFIGURE_COMMAND ""
|
||||
BUILD_COMMAND ""
|
||||
INSTALL_COMMAND ""
|
||||
TEST_COMMAND ""
|
||||
)
|
||||
|
||||
if(${CMAKE_VERSION} VERSION_LESS "3.3.0")
|
||||
set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/cub_dummy.c)
|
||||
file(WRITE ${dummyfile} "const char *dummy = \"${dummyfile}\";")
|
||||
add_library(cub STATIC ${dummyfile})
|
||||
else()
|
||||
add_library(cub INTERFACE)
|
||||
endif()
|
||||
|
||||
add_dependencies(cub extern_cub)
|
||||
|
||||
LIST(APPEND externl_project_dependencies cub)
|
||||
@ -0,0 +1,57 @@
|
||||
# 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.
|
||||
#
|
||||
|
||||
OPTION(WITH_LIBXSMM "Compile with libxsmm" OFF)
|
||||
|
||||
IF(NOT WITH_LIBXSMM)
|
||||
return()
|
||||
ENDIF()
|
||||
|
||||
IF(WIN32 OR APPLE OR ANDROID OR IOS)
|
||||
MESSAGE(WARNING "Windows, Mac or Mobile are not supported with libxsmm in Paddle yet.")
|
||||
SET(WITH_LIBXSMM OFF CACHE STRING "Disable LIBXSMM" FORCE)
|
||||
return()
|
||||
ENDIF()
|
||||
|
||||
INCLUDE (ExternalProject)
|
||||
|
||||
SET(LIBXSMM_SOURCES_DIR ${THIRD_PARTY_PATH}/libxsmm)
|
||||
SET(LIBXSMM_INSTALL_DIR ${THIRD_PARTY_PATH}/install/libxsmm)
|
||||
SET(LIBXSMM_INCLUDE_DIR "${LIBXSMM_INSTALL_DIR}/include" CACHE PATH "LIBXSMM include directory." FORCE)
|
||||
SET(LIBXSMM_LIBRARY_DIR "${LIBXSMM_INSTALL_DIR}/lib" CACHE PATH "LIBXSMM library directory." FORCE)
|
||||
SET(LIBXSMM_LIBS "${LIBXSMM_LIBRARY_DIR}/libxsmm.a"
|
||||
"${LIBXSMM_LIBRARY_DIR}/libxsmmnoblas.a")
|
||||
|
||||
ExternalProject_Add(
|
||||
extern_libxsmm
|
||||
GIT_REPOSITORY "https://github.com/hfp/libxsmm.git"
|
||||
GIT_TAG "7cc03b5b342fdbc6b6d990b190671c5dbb8489a2"
|
||||
PREFIX ${LIBXSMM_SOURCES_DIR}
|
||||
UPDATE_COMMAND ""
|
||||
CONFIGURE_COMMAND ""
|
||||
BUILD_IN_SOURCE 1
|
||||
BUILD_COMMAND $(MAKE) --silent PREFIX=${LIBXSMM_INSTALL_DIR} CXX=g++ CC=gcc WARP=0 install
|
||||
INSTALL_COMMAND ""
|
||||
)
|
||||
ADD_LIBRARY(libxsmm STATIC IMPORTED GLOBAL)
|
||||
SET_PROPERTY(TARGET libxsmm PROPERTY IMPORTED_LOCATION "${LIBXSMM_LIBRARY_DIR}/libxsmm.a")
|
||||
SET_PROPERTY(TARGET libxsmm PROPERTY IMPORTED_LOCATION "${LIBXSMM_LIBRARY_DIR}/libxsmmnoblas.a")
|
||||
|
||||
MESSAGE(STATUS "Libxsmm library: ${LIBXSMM_LIBS}")
|
||||
include_directories(${LIBXSMM_INCLUDE_DIR})
|
||||
ADD_DEFINITIONS(-DPADDLE_WITH_LIBXSMM)
|
||||
ADD_DEPENDENCIES(libxsmm extern_libxsmm)
|
||||
LIST(APPEND external_project_dependencies libxsmm)
|
||||
|
||||
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in new issue