使用CMAKE的cuda_compile_ptx编译成ptx

Question

文件kernel.cu中有一个简单的内核

__global__ void add1( double * pi, double c ) 
{
    *pi += c;
}

并可以轻松地将其编译到ptx文件kernel.ptx中：

nvcc -ptx kernel.cu

现在，我想使用以下CMakeLists.txt使用cmake再现相同的行为：

cmake_minimum_required(VERSION 2.8)
project(cmake_ptx)
find_package(CUDA REQUIRED)
cuda_compile_ptx(
  test
  kernel.cu
)

但当我打字的时候

克莱。&&制造

未生成ptx文件。如何使用cmake再现上述nvcc命令的行为？特别是如何：

• 将ptx文件的独立编译添加到目标(例如，所有)：我注意到，当同一个CMakeFile中有另一个cuda_add_executable时，它也构建ptx文件，否则不会。
• 将该文件命名为源文件，但使用.ptx而不是.cu结尾:与此问题相关：ptx in CMake?

Answer 1

cuda_compile_ptx只创建生成文件的规则，但不将它们添加到任何目标。您需要添加自定义目标，这取决于ptx文件：

cmake_minimum_required(VERSION 2.8)
project(cmake_ptx)
find_package(CUDA REQUIRED)
cuda_compile_ptx(
  cuda_ptx_files
  kernel.cu
)
add_custom_target(ptx ALL
    DEPENDS ${cuda_ptx_files} kernel.cu
    SOURCES kernel.cu)

现在，如果您运行make或make ptx，它将生成ptx文件。

Answer 2

仅供参考，这就是我的项目设置(我有一个src/和一个包含/文件夹)的工作原理，使用来自已接受的答案的提示：

cmake_minimum_required(VERSION 2.8)

project(cmake_ptx)

find_package(CUDA REQUIRED)
include_directories(include/)

cuda_compile_ptx(
  cuda_ptx_files
  src/common_kernels.cu
  OPTIONS -DCUDA_MATLAB
)
add_custom_command(OUTPUT ${CMAKE_BINARY_DIR}/common_kernels.ptx COMMAND ${CMAKE_COMMAND} -E rename ${cuda_ptx_files} ${CMAKE_BINARY_DIR}/common_kernels.ptx DEPENDS ${cuda_ptx_files}) 
add_custom_target(ptx ALL
    DEPENDS ${CMAKE_BINARY_DIR}/common_kernels.ptx src/common_kernels.cu
    SOURCES src/common_kernels.cu
)

这给了我与调用相同的输出：

nvcc -ptx src/include_kernels.cu -I -I/ -DCUDA_MATLAB