TransferBench Package

Author: PedramAlizadeh

CMakeLists.txt

@@ -5,6 +5,7 @@ else()
     set(ROCM_PATH "/opt/rocm" CACHE STRING "ROCm install directory")
 endif()
 cmake_minimum_required(VERSION 3.5)
+set( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -pthread")
 project(TransferBench CXX)
 set( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 -L${ROCM_PATH}/hsa/lib")
 include_directories(${ROCM_PATH}/hsa/include)
@@ -13,3 +14,35 @@ set (CMAKE_RUNTIME_OUTPUT_DIRECTORY ..)
 add_executable(TransferBench src/TransferBench.cpp)
 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_package_setup_component(${PACKAGE_NAME}
+    LIBRARY_NAME ${LIBRARY_NAME}
+    PACKAGE_NAME ${PACKAGE_NAME}
+)
+
+# rocm_install(DIRECTORY ${PROJECT_BINARY_DIR}/src/include
+# DESTINATION include/${PACKAGE_NAME}/build
+# COMPONENT ${LIBRARY_NAME}-headers
+# FILES_MATCHING
+# PATTERN "*.hpp"
+# )
+
+rocm_install(DIRECTORY src/include
+DESTINATION include/${PACKAGE_NAME}
+COMPONENT ${LIBRARY_NAME}
+FILES_MATCHING
+PATTERN "*.hpp"
+)
+
+rocm_create_package(
+    NAME ${LIBRARY_NAME}
+    DESCRIPTION "TransferBench package"
+    MAINTAINER "RCCL Team"
+)