Merge branch 'develop' of https://github.com/PaddlePaddle/Paddle into add_py36_py37_ubuntu_dockerfile
	
		
	
				
					
				
			
						commit
						e7480c1c8d
					
				@ -0,0 +1,31 @@
 | 
				
			||||
include(ExternalProject)
 | 
				
			||||
 | 
				
			||||
set(DLPACK_SOURCE_DIR ${THIRD_PARTY_PATH}/dlpack)
 | 
				
			||||
set(DLPACK_INCLUDE_DIR ${DLPACK_SOURCE_DIR}/src/extern_dlpack/include)
 | 
				
			||||
 | 
				
			||||
include_directories(${DLPACK_INCLUDE_DIR})
 | 
				
			||||
 | 
				
			||||
ExternalProject_Add(
 | 
				
			||||
  extern_dlpack
 | 
				
			||||
  ${EXTERNAL_PROJECT_LOG_ARGS}
 | 
				
			||||
  GIT_REPOSITORY "https://github.com/dmlc/dlpack.git"
 | 
				
			||||
  GIT_TAG        "v0.2"
 | 
				
			||||
  PREFIX         ${DLPACK_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}/dlpack_dummy.c)
 | 
				
			||||
  file(WRITE ${dummyfile} "const char *dummy = \"${dummyfile}\";")
 | 
				
			||||
  add_library(dlpack STATIC ${dummyfile})
 | 
				
			||||
else()
 | 
				
			||||
  add_library(dlpack INTERFACE)
 | 
				
			||||
endif()
 | 
				
			||||
 | 
				
			||||
add_dependencies(dlpack extern_dlpack)
 | 
				
			||||
 | 
				
			||||
LIST(APPEND externl_project_dependencies dlpack)
 | 
				
			||||
@ -0,0 +1,44 @@
 | 
				
			||||
if (NOT WITH_AMD_GPU)
 | 
				
			||||
    return()
 | 
				
			||||
endif()
 | 
				
			||||
 | 
				
			||||
# rocprim is "ROCm Parallel Primitives" for short.
 | 
				
			||||
# It is a header-only library providing HIP and HC parallel primitives
 | 
				
			||||
# for developing performant GPU-accelerated code on AMD ROCm platform.
 | 
				
			||||
 | 
				
			||||
if("x${HCC_HOME}" STREQUAL "x")
 | 
				
			||||
  set(HCC_HOME "/opt/rocm/hcc")
 | 
				
			||||
endif()
 | 
				
			||||
 | 
				
			||||
INCLUDE(ExternalProject)
 | 
				
			||||
 | 
				
			||||
SET(ROCPRIM_SOURCE_DIR ${THIRD_PARTY_PATH}/rocprim)
 | 
				
			||||
SET(ROCPRIM_INSTALL_DIR  ${THIRD_PARTY_PATH}/install/rocprim)
 | 
				
			||||
SET(ROCPRIM_INCLUDE_DIR ${ROCPRIM_INSTALL_DIR}/include)
 | 
				
			||||
 | 
				
			||||
ExternalProject_Add(
 | 
				
			||||
    extern_rocprim
 | 
				
			||||
    GIT_REPOSITORY "https://github.com/ROCmSoftwarePlatform/rocPRIM.git"
 | 
				
			||||
    GIT_TAG        5bd41b96ab8d8343330fb2c3e1b96775bde3b3fc 
 | 
				
			||||
    PREFIX         ${ROCPRIM_SOURCE_DIR}
 | 
				
			||||
    UPDATE_COMMAND  ""
 | 
				
			||||
    CMAKE_ARGS     -DCMAKE_CXX_COMPILER=${HCC_HOME}/bin/hcc
 | 
				
			||||
    CMAKE_ARGS     -DONLY_INSTALL=ON
 | 
				
			||||
    CMAKE_ARGS     -DBUILD_TEST=OFF
 | 
				
			||||
    CMAKE_ARGS     -DCMAKE_INSTALL_PREFIX=${ROCPRIM_INSTALL_DIR}
 | 
				
			||||
 | 
				
			||||
    INSTALL_DIR    ${ROCPRIM_INSTALL_DIR}
 | 
				
			||||
    ${EXTERNAL_PROJECT_LOG_ARGS}
 | 
				
			||||
)
 | 
				
			||||
 | 
				
			||||
INCLUDE_DIRECTORIES(${ROCPRIM_INCLUDE_DIR})
 | 
				
			||||
 | 
				
			||||
if (${CMAKE_VERSION} VERSION_LESS "3.3.0")
 | 
				
			||||
    set(dummyfile ${CMAKE_CURRENT_BINARY_DIR}/rocprim_dummy.c)
 | 
				
			||||
    file(WRITE ${dummyfile} "const char *dummy_rocprim = \"${dummyfile}\";")
 | 
				
			||||
    add_library(rocprim STATIC ${dummyfile})
 | 
				
			||||
else()
 | 
				
			||||
    add_library(rocprim INTERFACE)
 | 
				
			||||
endif()
 | 
				
			||||
 | 
				
			||||
add_dependencies(rocprim extern_rocprim)
 | 
				
			||||
Some files were not shown because too many files have changed in this diff Show More
					Loading…
					
					
				
		Reference in new issue