mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2025-04-27 18:21:30 +01:00
38 lines
1020 B
CMake
38 lines
1020 B
CMake
# Include directories and libraries
|
|
include_directories(../../../Common)
|
|
|
|
# Source file
|
|
set(SRC_FILES
|
|
ptxjit.cpp
|
|
)
|
|
|
|
# Add target for ptxjit
|
|
add_executable(ptxjit ${SRC_FILES})
|
|
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} ${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 matrixMulDrv depends on the fatbin
|
|
add_dependencies(matrixMulDrv generate_ptxjit_ptx)
|