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.
75 lines
2.5 KiB
75 lines
2.5 KiB
# MIT License |
|
# |
|
# Copyright (c) 2022-2024 Advanced Micro Devices, Inc. All rights reserved. |
|
# |
|
# Permission is hereby granted, free of charge, to any person obtaining a copy |
|
# of this software and associated documentation files (the "Software"), to deal |
|
# in the Software without restriction, including without limitation the rights |
|
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell |
|
# copies of the Software, and to permit persons to whom the Software is |
|
# furnished to do so, subject to the following conditions: |
|
# |
|
# The above copyright notice and this permission notice shall be included in all |
|
# copies or substantial portions of the Software. |
|
# |
|
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR |
|
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, |
|
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE |
|
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER |
|
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, |
|
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE |
|
# SOFTWARE. |
|
|
|
set(example_name rocrand_simple_distributions_cpp) |
|
|
|
cmake_minimum_required(VERSION 3.21 FATAL_ERROR) |
|
project(${example_name} LANGUAGES CXX) |
|
|
|
set(GPU_RUNTIME "HIP" CACHE STRING "Switches between HIP and CUDA") |
|
set(GPU_RUNTIMES "HIP" "CUDA") |
|
set_property(CACHE GPU_RUNTIME PROPERTY STRINGS ${GPU_RUNTIMES}) |
|
|
|
if(NOT "${GPU_RUNTIME}" IN_LIST GPU_RUNTIMES) |
|
message( |
|
FATAL_ERROR |
|
"Only the following values are accepted for GPU_RUNTIME: ${GPU_RUNTIMES}" |
|
) |
|
endif() |
|
|
|
if(GPU_RUNTIME STREQUAL "CUDA") |
|
set(LANG "CUDA") |
|
enable_language(CUDA) |
|
else() |
|
set(LANG "CXX") |
|
endif() |
|
|
|
set(CMAKE_${LANG}_STANDARD 17) |
|
set(CMAKE_${LANG}_EXTENSIONS OFF) |
|
set(CMAKE_${LANG}_STANDARD_REQUIRED ON) |
|
|
|
if(NOT CMAKE_PREFIX_PATH) |
|
set(CMAKE_PREFIX_PATH "/opt/rocm") |
|
endif() |
|
|
|
find_package(rocrand REQUIRED) |
|
|
|
add_executable(${example_name} main.cpp) |
|
add_test(${example_name} ${example_name}) |
|
|
|
if(GPU_RUNTIME STREQUAL "CUDA") |
|
target_link_libraries(${example_name} PRIVATE roc::rocrand) |
|
set_source_files_properties(main.cpp PROPERTIES LANGUAGE CUDA) |
|
else() |
|
target_link_libraries(${example_name} roc::rocrand hip::host) |
|
endif() |
|
|
|
target_include_directories(${example_name} PRIVATE "../../../Common") |
|
if(WIN32) |
|
target_compile_definitions(${example_name} PRIVATE WIN32) |
|
endif() |
|
|
|
install(TARGETS ${example_name}) |
|
|
|
if(CMAKE_SYSTEM_NAME MATCHES Windows) |
|
install(IMPORTED_RUNTIME_ARTIFACTS roc::rocrand) |
|
endif()
|
|
|