2fb764d2da
Summary: We use `add_llvm_library` as a shorthand for setting up all the dependencies and libraries we need for the OpenMP offloading runtime as they depend on a lot of the LLVM utilities. However, we always explicitly installed these manually. Behind the scenes the function would then install it again. This was unnoticed because until now the destinations matched. Now that we want it to optionally go into the other directory it is duplicating them. Fix this by stating that this is a build tree only library so we can handle it ourselves.
89 lines
3.5 KiB
CMake
89 lines
3.5 KiB
CMake
##===----------------------------------------------------------------------===##
|
|
#
|
|
# Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
|
|
# See https://llvm.org/LICENSE.txt for license information.
|
|
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
#
|
|
##===----------------------------------------------------------------------===##
|
|
#
|
|
# Build a plugin for a CUDA machine if available.
|
|
#
|
|
##===----------------------------------------------------------------------===##
|
|
set(LIBOMPTARGET_BUILD_CUDA_PLUGIN TRUE CACHE BOOL
|
|
"Whether to build CUDA plugin")
|
|
if (NOT LIBOMPTARGET_BUILD_CUDA_PLUGIN)
|
|
libomptarget_say("Not building CUDA NextGen offloading plugin: LIBOMPTARGET_BUILD_CUDA_PLUGIN is false")
|
|
return()
|
|
endif()
|
|
|
|
if (NOT (CMAKE_SYSTEM_PROCESSOR MATCHES "(x86_64)|(ppc64le)|(aarch64)$" AND CMAKE_SYSTEM_NAME MATCHES "Linux"))
|
|
libomptarget_say("Not building CUDA NextGen offloading plugin: only support CUDA in Linux x86_64, ppc64le, or aarch64 hosts.")
|
|
return()
|
|
endif()
|
|
|
|
libomptarget_say("Building CUDA NextGen offloading plugin.")
|
|
|
|
set(LIBOMPTARGET_DLOPEN_LIBCUDA OFF)
|
|
option(LIBOMPTARGET_FORCE_DLOPEN_LIBCUDA "Build with dlopened libcuda" ${LIBOMPTARGET_DLOPEN_LIBCUDA})
|
|
|
|
add_llvm_library(omptarget.rtl.cuda SHARED
|
|
src/rtl.cpp
|
|
|
|
LINK_COMPONENTS
|
|
Support
|
|
Object
|
|
|
|
LINK_LIBS PRIVATE
|
|
PluginCommon
|
|
${OPENMP_PTHREAD_LIB}
|
|
|
|
NO_INSTALL_RPATH
|
|
BUILDTREE_ONLY
|
|
)
|
|
|
|
if ((OMPT_TARGET_DEFAULT) AND (LIBOMPTARGET_OMPT_SUPPORT))
|
|
target_link_libraries(omptarget.rtl.cuda PRIVATE OMPT)
|
|
endif()
|
|
|
|
if (LIBOMP_HAVE_VERSION_SCRIPT_FLAG)
|
|
target_link_libraries(omptarget.rtl.cuda PRIVATE
|
|
"-Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/../exports,-z,defs")
|
|
endif()
|
|
|
|
|
|
if(LIBOMPTARGET_DEP_CUDA_FOUND AND NOT LIBOMPTARGET_FORCE_DLOPEN_LIBCUDA)
|
|
libomptarget_say("Building CUDA plugin linked against libcuda")
|
|
target_link_libraries(omptarget.rtl.cuda PRIVATE CUDA::cuda_driver)
|
|
else()
|
|
libomptarget_say("Building CUDA plugin for dlopened libcuda")
|
|
target_include_directories(omptarget.rtl.cuda PRIVATE dynamic_cuda)
|
|
target_sources(omptarget.rtl.cuda PRIVATE dynamic_cuda/cuda.cpp)
|
|
endif()
|
|
|
|
# Define debug prefix. TODO: This should be automatized in the Debug.h but it
|
|
# requires changing the original plugins.
|
|
target_compile_definitions(omptarget.rtl.cuda PRIVATE TARGET_NAME="CUDA")
|
|
target_compile_definitions(omptarget.rtl.cuda PRIVATE DEBUG_PREFIX="TARGET CUDA RTL")
|
|
|
|
target_include_directories(omptarget.rtl.cuda PRIVATE ${LIBOMPTARGET_INCLUDE_DIR})
|
|
|
|
# Configure testing for the CUDA plugin. We will build tests if we could a
|
|
# functional NVIDIA GPU on the system, or if manually specifies by the user.
|
|
option(LIBOMPTARGET_FORCE_NVIDIA_TESTS "Build NVIDIA libomptarget tests" OFF)
|
|
if (LIBOMPTARGET_FOUND_NVIDIA_GPU OR LIBOMPTARGET_FORCE_NVIDIA_TESTS)
|
|
libomptarget_say("Enable tests using CUDA plugin")
|
|
set(LIBOMPTARGET_SYSTEM_TARGETS
|
|
"${LIBOMPTARGET_SYSTEM_TARGETS} nvptx64-nvidia-cuda nvptx64-nvidia-cuda-LTO" PARENT_SCOPE)
|
|
list(APPEND LIBOMPTARGET_TESTED_PLUGINS "omptarget.rtl.cuda")
|
|
set(LIBOMPTARGET_TESTED_PLUGINS "${LIBOMPTARGET_TESTED_PLUGINS}" PARENT_SCOPE)
|
|
else()
|
|
libomptarget_say("Not generating NVIDIA tests, no supported devices detected."
|
|
" Use 'LIBOMPTARGET_FORCE_NVIDIA_TESTS' to override.")
|
|
endif()
|
|
|
|
# Install plugin under the lib destination folder.
|
|
install(TARGETS omptarget.rtl.cuda LIBRARY DESTINATION "${OPENMP_INSTALL_LIBDIR}")
|
|
set_target_properties(omptarget.rtl.cuda PROPERTIES
|
|
INSTALL_RPATH "$ORIGIN" BUILD_RPATH "$ORIGIN:${CMAKE_CURRENT_BINARY_DIR}/.."
|
|
CXX_VISIBILITY_PRESET protected)
|