derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
73 lines
2.6 KiB
CMake
73 lines
2.6 KiB
CMake
# Copyright (c) Advanced Micro Devices, Inc.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
check_rocprofsys_disable_examples("matrix-exponential" _MATRIX_EXPONENTIAL_DISABLED)
|
|
check_rocprofsys_disable_examples("openmp" _OPENMP_DISABLED)
|
|
if(_MATRIX_EXPONENTIAL_DISABLED OR _OPENMP_DISABLED)
|
|
return()
|
|
endif()
|
|
|
|
if(NOT HIPCC_EXECUTABLE)
|
|
rocprofiler_systems_message(AUTHOR_WARNING "hipcc not found. Cannot build matrix-exponential-streams-sync-hip target.")
|
|
return()
|
|
endif()
|
|
|
|
if(NOT CMAKE_CXX_COMPILER_IS_HIPCC)
|
|
if(
|
|
CMAKE_CXX_COMPILER STREQUAL HIPCC_EXECUTABLE
|
|
OR "${CMAKE_CXX_COMPILER}" MATCHES "hipcc"
|
|
)
|
|
set(CMAKE_CXX_COMPILER_IS_HIPCC 1 CACHE BOOL "HIP compiler")
|
|
endif()
|
|
endif()
|
|
|
|
find_package(OpenMP QUIET COMPONENTS CXX)
|
|
if(NOT OpenMP_CXX_FOUND)
|
|
rocprofiler_systems_message(AUTHOR_WARNING "OpenMP not found, this is required for matrix-exponential example")
|
|
return()
|
|
endif()
|
|
|
|
find_library(ROCBLAS_LIBRARY NAMES rocblas PATHS ${ROCM_PATH}/lib ${ROCM_PATH}/lib64)
|
|
if(NOT ROCBLAS_LIBRARY)
|
|
rocprofiler_systems_message(AUTHOR_WARNING "rocblas not found, this is required for matrix-exponential example")
|
|
return()
|
|
endif()
|
|
|
|
find_library(
|
|
ROCTX_SDK_LIBRARY
|
|
NAMES rocprofiler-sdk-roctx
|
|
PATHS ${ROCM_PATH}/lib ${ROCM_PATH}/lib64
|
|
)
|
|
if(NOT ROCTX_SDK_LIBRARY)
|
|
rocprofiler_systems_message(AUTHOR_WARNING "rocprofiler-sdk-roctx not found, this is required for matrix-exponential example")
|
|
return()
|
|
endif()
|
|
|
|
set(HPC_EXAMPLE_NAME "matrix-exponential-streams-sync-hip")
|
|
add_executable(${HPC_EXAMPLE_NAME} mat-exp.cpp)
|
|
|
|
# Derive the rocblas include directory from the library path so that the
|
|
# compiler can find <rocblas/rocblas.h> regardless of which ROCm tree the
|
|
# library was found in (system /opt/rocm vs TheRock dist).
|
|
get_filename_component(_rocblas_lib_dir "${ROCBLAS_LIBRARY}" DIRECTORY)
|
|
get_filename_component(_rocblas_prefix "${_rocblas_lib_dir}" DIRECTORY)
|
|
|
|
target_compile_options(
|
|
${HPC_EXAMPLE_NAME}
|
|
PRIVATE ${OpenMP_CXX_FLAGS} -Wno-vla-cxx-extension
|
|
)
|
|
target_include_directories(${HPC_EXAMPLE_NAME} PRIVATE "${_rocblas_prefix}/include")
|
|
target_link_options(${HPC_EXAMPLE_NAME} PRIVATE -fopenmp)
|
|
target_link_libraries(${HPC_EXAMPLE_NAME} PRIVATE ${ROCBLAS_LIBRARY} ${ROCTX_SDK_LIBRARY})
|
|
|
|
foreach(arch IN LISTS ROCPROFSYS_GFX_TARGETS)
|
|
target_compile_options(${HPC_EXAMPLE_NAME} PRIVATE --offload-arch=${arch})
|
|
target_link_options(${HPC_EXAMPLE_NAME} PRIVATE --offload-arch=${arch})
|
|
endforeach()
|
|
|
|
if(NOT CMAKE_CXX_COMPILER_IS_HIPCC)
|
|
rocprofiler_systems_custom_compilation(COMPILER ${HIPCC_EXECUTABLE} TARGET ${HPC_EXAMPLE_NAME})
|
|
endif()
|
|
|
|
rocprofiler_systems_install_hpc_example(${HPC_EXAMPLE_NAME})
|