| ##===----------------------------------------------------------------------===## |
| # |
| # 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.") |
| |
| # Define the suffix for the runtime messaging dumps. |
| add_definitions("-DTARGET_NAME=CUDA") |
| |
| # Define debug prefix. TODO: This should be automatized in the Debug.h but it |
| # requires changing the original plugins. |
| add_definitions(-DDEBUG_PREFIX="TARGET CUDA RTL") |
| |
| set(LIBOMPTARGET_DLOPEN_LIBCUDA OFF) |
| option(LIBOMPTARGET_FORCE_DLOPEN_LIBCUDA "Build with dlopened libcuda" ${LIBOMPTARGET_DLOPEN_LIBCUDA}) |
| |
| set(LIBOMPTARGET_CAN_LINK_LIBCUDA FALSE) |
| if (LIBOMPTARGET_DEP_CUDA_FOUND AND LIBOMPTARGET_DEP_CUDA_DRIVER_FOUND) |
| set(LIBOMPTARGET_CAN_LINK_LIBCUDA TRUE) |
| endif() |
| |
| if (LIBOMPTARGET_CAN_LINK_LIBCUDA AND NOT LIBOMPTARGET_FORCE_DLOPEN_LIBCUDA) |
| libomptarget_say("Building CUDA NextGen plugin linked against libcuda") |
| include_directories(${LIBOMPTARGET_DEP_CUDA_INCLUDE_DIRS}) |
| add_llvm_library(omptarget.rtl.cuda.nextgen SHARED |
| |
| src/rtl.cpp |
| |
| ADDITIONAL_HEADER_DIRS |
| ${LIBOMPTARGET_INCLUDE_DIR} |
| |
| LINK_LIBS |
| PRIVATE |
| elf_common |
| MemoryManager |
| PluginInterface |
| ${LIBOMPTARGET_DEP_CUDA_DRIVER_LIBRARIES} |
| ${OPENMP_PTHREAD_LIB} |
| "-Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/../exports" |
| "-Wl,-z,defs" |
| |
| NO_INSTALL_RPATH |
| ) |
| else() |
| libomptarget_say("Building CUDA NextGen plugin for dlopened libcuda") |
| include_directories(../../plugins/cuda/dynamic_cuda) |
| add_llvm_library(omptarget.rtl.cuda.nextgen |
| SHARED |
| |
| src/rtl.cpp |
| ../../plugins/cuda/dynamic_cuda/cuda.cpp |
| |
| ADDITIONAL_HEADER_DIRS |
| ${LIBOMPTARGET_INCLUDE_DIR} |
| |
| LINK_LIBS |
| PRIVATE |
| elf_common |
| MemoryManager |
| PluginInterface |
| ${CMAKE_DL_LIBS} |
| ${OPENMP_PTHREAD_LIB} |
| "-Wl,--version-script=${CMAKE_CURRENT_SOURCE_DIR}/../exports" |
| "-Wl,-z,defs" |
| |
| NO_INSTALL_RPATH |
| ) |
| endif() |
| |
| # Install plugin under the lib destination folder. |
| install(TARGETS omptarget.rtl.cuda.nextgen LIBRARY DESTINATION "${OPENMP_INSTALL_LIBDIR}") |
| set_target_properties(omptarget.rtl.cuda.nextgen PROPERTIES |
| INSTALL_RPATH "$ORIGIN" BUILD_RPATH "$ORIGIN:${CMAKE_CURRENT_BINARY_DIR}/.." |
| CXX_VISIBILITY_PRESET protected) |
| |
| target_include_directories(omptarget.rtl.cuda.nextgen PRIVATE |
| ${LIBOMPTARGET_INCLUDE_DIR} |
| ) |