You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
66 lines
2.2 KiB
66 lines
2.2 KiB
if(NOT WITH_AMD_GPU)
|
|
return()
|
|
endif()
|
|
|
|
include_directories("/opt/rocm/include")
|
|
include_directories("/opt/rocm/hip/include")
|
|
include_directories("/opt/rocm/miopen/include")
|
|
include_directories("/opt/rocm/hipblas/include")
|
|
include_directories("/opt/rocm/hiprand/include")
|
|
include_directories("/opt/rocm/rocrand/include")
|
|
include_directories("/opt/rocm/rccl/include")
|
|
include_directories("/opt/rocm/thrust")
|
|
|
|
list(APPEND EXTERNAL_LIBS "-L/opt/rocm/lib/ -lhip_hcc")
|
|
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -fPIC -DPADDLE_WITH_HIP -std=c++11" )
|
|
|
|
if(WITH_DSO)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_USE_DSO")
|
|
endif(WITH_DSO)
|
|
|
|
if(WITH_TESTING)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_WITH_TESTING")
|
|
endif(WITH_TESTING)
|
|
|
|
if(WITH_DISTRIBUTE)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_WITH_DISTRIBUTE")
|
|
endif(WITH_DISTRIBUTE)
|
|
|
|
if(WITH_GRPC)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_WITH_GRPC")
|
|
endif(WITH_GRPC)
|
|
|
|
if(NOT WITH_GOLANG)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_WITHOUT_GOLANG")
|
|
endif(NOT WITH_GOLANG)
|
|
|
|
if(WITH_MKLDNN)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_WITH_MKLDNN")
|
|
endif(WITH_MKLDNN)
|
|
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DANY_IMPL_ANY_CAST_MOVEABLE")
|
|
|
|
if(NOT WITH_RDMA)
|
|
set(HIP_HCC_FLAGS "${HIP_HCC_FLAGS} -DPADDLE_DISABLE_RDMA")
|
|
endif(NOT WITH_RDMA)
|
|
|
|
|
|
|
|
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
|
list(APPEND HIP_HCC_FLAGS ${CMAKE_CXX_FLAGS_DEBUG})
|
|
elseif(CMAKE_BUILD_TYPE STREQUAL "RelWithDebInfo")
|
|
list(APPEND HIP_HCC_FLAGS ${CMAKE_CXX_FLAGS_RELWITHDEBINFO})
|
|
elseif(CMAKE_BUILD_TYPE STREQUAL "MinSizeRel")
|
|
list(APPEND HIP_HCC_FLAGS ${CMAKE_CXX_FLAGS_MINSIZEREL})
|
|
endif()
|
|
|
|
if("x${HCC_HOME}" STREQUAL "x")
|
|
set(HCC_HOME "/opt/rocm/hcc")
|
|
endif()
|
|
|
|
set(CMAKE_HIP_LINK_EXECUTABLE "${HIP_HIPCC_CMAKE_LINKER_HELPER} ${HCC_HOME} <FLAGS> <CMAKE_CXX_LINK_FLAGS> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES>")
|
|
set(CMAKE_HIP_CREATE_SHARED_LIBRARY "${HIP_HIPCC_CMAKE_LINKER_HELPER} ${HCC_HOME} <CMAKE_CXX_LINK_FLAGS> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES> -shared")
|
|
set(CMAKE_HIP_CREATE_SHARED_MODULE "${HIP_HIPCC_CMAKE_LINKER_HELPER} ${HCC_HOME} <CMAKE_CXX_LINK_FLAGS> <LINK_FLAGS> <OBJECTS> -o <TARGET> <LINK_LIBRARIES> -shared")
|
|
|