| ##===----------------------------------------------------------------------===## |
| # |
| # 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 |
| ) |
| |
| 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) |