CMakeLists.txt 1.12 KB
Newer Older
1
# Copyright (c) 2023-2024 Advanced Micro Devices, Inc. All rights reserved.
2

PedramAlizadeh's avatar
PedramAlizadeh committed
3
4
5
6
7
8
if (DEFINED ENV{ROCM_PATH})
    set(ROCM_PATH "$ENV{ROCM_PATH}" CACHE STRING "ROCm install directory")
else()
    set(ROCM_PATH "/opt/rocm" CACHE STRING "ROCm install directory")
endif()
cmake_minimum_required(VERSION 3.5)
9

10
11
project(TransferBench VERSION 1.54.0 LANGUAGES CXX)
set( CMAKE_CXX_FLAGS "${CMAKE_CXX_FLAGS} -O3 --std=c++20 -L${ROCM_PATH}/lib")
PedramAlizadeh's avatar
PedramAlizadeh committed
12
13
include_directories(${ROCM_PATH}/include)
link_libraries(numa hsa-runtime64 pthread)
14
15
16
set (CMAKE_RUNTIME_OUTPUT_DIRECTORY .)
add_executable(TransferBench src/client/Client.cpp)
target_include_directories(TransferBench PRIVATE src/header src/client src/client/Presets)
PedramAlizadeh's avatar
PedramAlizadeh committed
17

PedramAlizadeh's avatar
PedramAlizadeh committed
18
19
20
21
22
23
24
25
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)

26
rocm_install(TARGETS TransferBench COMPONENT devel)
PedramAlizadeh's avatar
PedramAlizadeh committed
27

28
rocm_package_add_dependencies(DEPENDS numactl hsa-rocr)
29

PedramAlizadeh's avatar
PedramAlizadeh committed
30
31
32
33
34
rocm_create_package(
    NAME ${LIBRARY_NAME}
    DESCRIPTION "TransferBench package"
    MAINTAINER "RCCL Team <gilbert.lee@amd.com>"
)