bd6df0fe21
Summary: There were a few issues with the first one, leading to some errors and warnings. Most importantly, this was building on MSVC which isn't supported.
47 lines
1.4 KiB
CMake
47 lines
1.4 KiB
CMake
set(LLVM_LINK_COMPONENTS
|
|
BinaryFormat
|
|
Object
|
|
Option
|
|
Support
|
|
FrontendOffloading
|
|
TargetParser
|
|
)
|
|
|
|
add_llvm_tool(llvm-gpu-loader
|
|
llvm-gpu-loader.cpp
|
|
|
|
# TODO: We intentionally split this currently due to statically linking the
|
|
# GPU runtimes. Dynamically load the dependencies, possibly using the
|
|
# LLVM offloading API when it is complete.
|
|
PARTIAL_SOURCES_INTENDED
|
|
|
|
DEPENDS
|
|
intrinsics_gen
|
|
)
|
|
|
|
# Locate the RPC server handling interface.
|
|
include(FindLibcCommonUtils)
|
|
target_link_libraries(llvm-gpu-loader PUBLIC llvm-libc-common-utilities)
|
|
|
|
# Check for HSA support for targeting AMD GPUs.
|
|
find_package(hsa-runtime64 QUIET 1.2.0 HINTS ${CMAKE_INSTALL_PREFIX} PATHS /opt/rocm)
|
|
if(hsa-runtime64_FOUND)
|
|
target_sources(llvm-gpu-loader PRIVATE amdhsa.cpp)
|
|
target_compile_definitions(llvm-gpu-loader PRIVATE AMDHSA_SUPPORT)
|
|
target_link_libraries(llvm-gpu-loader PRIVATE hsa-runtime64::hsa-runtime64)
|
|
|
|
# Compatibility with the old amdhsa-loader name.
|
|
add_llvm_tool_symlink(amdhsa-loader llvm-gpu-loader)
|
|
endif()
|
|
|
|
# Check for CUDA support for targeting NVIDIA GPUs.
|
|
find_package(CUDAToolkit 11.2 QUIET)
|
|
if(CUDAToolkit_FOUND)
|
|
target_sources(llvm-gpu-loader PRIVATE nvptx.cpp)
|
|
target_compile_definitions(llvm-gpu-loader PRIVATE NVPTX_SUPPORT)
|
|
target_link_libraries(llvm-gpu-loader PRIVATE CUDA::cuda_driver)
|
|
|
|
# Compatibility with the old nvptx-loader name.
|
|
add_llvm_tool_symlink(nvptx-loader llvm-gpu-loader)
|
|
endif()
|