Skip to content

GitLab

Unverified Commit e3781d73 authored by Pedram Alizadeh's avatar Pedram Alizadeh Committed by GitHub
Browse files

Merge pull request #12 from PedramAlizadeh/packaging 

TransferBench Package
parents 068c5ce1 5b366fa2
Show whitespace changes
Inline Side-by-side
Showing with 17 additions and 1 deletion
CMakeLists.txt
View file @ e3781d73
...@@ -5,7 +5,8 @@ else() ...@@ -5,7 +5,8 @@ else()
set(ROCM_PATH "/opt/rocm" CACHE STRING "ROCm install directory") set(ROCM_PATH "/opt/rocm" CACHE STRING "ROCm install directory")
endif() endif()
cmake_minimum_required(VERSION 3.5) cmake_minimum_required(VERSION 3.5)
project(TransferBench CXX)
project(TransferBench VERSION 1.0 LANGUAGES CXX)
set( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 -L${ROCM_PATH}/lib") set( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 -L${ROCM_PATH}/lib")
include_directories(${ROCM_PATH}/include) include_directories(${ROCM_PATH}/include)
link_libraries(numa hsa-runtime64 pthread) link_libraries(numa hsa-runtime64 pthread)
...@@ -13,3 +14,18 @@ set (CMAKE_RUNTIME_OUTPUT_DIRECTORY ..) ...@@ -13,3 +14,18 @@ set (CMAKE_RUNTIME_OUTPUT_DIRECTORY ..)
add_executable(TransferBench src/TransferBench.cpp) add_executable(TransferBench src/TransferBench.cpp)
target_include_directories(TransferBench PRIVATE src/include) target_include_directories(TransferBench PRIVATE src/include)
find_package(ROCM 0.8 REQUIRED PATHS ${ROCM_PATH})
include(ROCMInstallTargets)
include(ROCMCreatePackage)
set(ROCMCHECKS_WARN_TOOLCHAIN_VAR OFF)
set(PACKAGE_NAME TB)
set(LIBRARY_NAME TransferBench)
rocm_install(TARGETS TransferBench)
rocm_create_package(
NAME ${LIBRARY_NAME}
DESCRIPTION "TransferBench package"
MAINTAINER "RCCL Team <gilbert.lee@amd.com>"
)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment