mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2025-04-04 07:21:33 +01:00
55 lines
1.6 KiB
CMake
55 lines
1.6 KiB
CMake
cmake_minimum_required(VERSION 3.20)
|
|
|
|
list(APPEND CMAKE_MODULE_PATH "${CMAKE_CURRENT_SOURCE_DIR}/../../../cmake/Modules")
|
|
|
|
project(ptxjit LANGUAGES C CXX CUDA)
|
|
|
|
find_package(CUDAToolkit REQUIRED)
|
|
|
|
set(CMAKE_POSITION_INDEPENDENT_CODE ON)
|
|
|
|
set(CMAKE_CUDA_ARCHITECTURES 50 52 60 61 70 72 75 80 86 87 89 90 100 101 120)
|
|
set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -Wno-deprecated-gpu-targets")
|
|
if(CMAKE_BUILD_TYPE STREQUAL "Debug")
|
|
# set(CMAKE_CUDA_FLAGS "${CMAKE_CUDA_FLAGS} -G") # enable cuda-gdb (expensive)
|
|
endif()
|
|
|
|
# Include directories and libraries
|
|
include_directories(../../../Common)
|
|
|
|
# Source file
|
|
# Add target for ptxjit
|
|
add_executable(ptxjit ptxjit.cpp)
|
|
|
|
target_compile_options(ptxjit PRIVATE $<$<COMPILE_LANGUAGE:CUDA>:--extended-lambda>)
|
|
|
|
target_compile_features(ptxjit PRIVATE cxx_std_17 cuda_std_17)
|
|
|
|
set_target_properties(ptxjit PROPERTIES CUDA_SEPARABLE_COMPILATION ON)
|
|
|
|
target_include_directories(ptxjit PRIVATE
|
|
${CUDAToolkit_INCLUDE_DIRS}
|
|
)
|
|
|
|
target_link_libraries(ptxjit PUBLIC
|
|
CUDA::cuda_driver
|
|
CUDA::cudart
|
|
)
|
|
|
|
set(CUDA_PTX_FILE "${CMAKE_CURRENT_BINARY_DIR}/ptxjit_kernel64.ptx")
|
|
set(CUDA_KERNEL_SOURCE "${CMAKE_CURRENT_SOURCE_DIR}/ptxjit_kernel.cu")
|
|
|
|
add_custom_command(
|
|
OUTPUT ${CUDA_PTX_FILE}
|
|
COMMAND ${CMAKE_CUDA_COMPILER} ${INCLUDES} ${ALL_CCFLAGS} -Wno-deprecated-gpu-targets ${GENCODE_FLAGS} -o ${CUDA_PTX_FILE} -ptx ${CUDA_KERNEL_SOURCE}
|
|
DEPENDS ${CUDA_KERNEL_SOURCE}
|
|
COMMENT "Building CUDA PTX: ${CUDA_PTX_FILE}"
|
|
)
|
|
|
|
|
|
# Create a dummy target for fatbin generation
|
|
add_custom_target(generate_ptxjit_ptx ALL DEPENDS ${CUDA_PTX_FILE})
|
|
|
|
# Ensure ptxjit depends on the fatbin
|
|
add_dependencies(ptxjit generate_ptxjit_ptx)
|