cuda - 如何在 CMake 中更改 cuda_compile_ptx 的输出文件名？

Question

在带有 FindCUDA 的 CMAKE 中，给定一个输入文件 filename.cu , cuda_compile_ptx命令生成 cuda_compile_ptx_generated_filename.cu.ptx 形式的输出文件名，但我需要输出文件名的形式为 filename.ptx .

有没有一种简单的方法可以使这项工作？

Answer 1

理想情况下，您应该能够指定 -o <outputName> .然而问题是CUDA_COMPILE_PTX宏实际上覆盖了 -o选项与 cuda_compile_ptx_generated_${filename}.ptx
这是将 CUDA 编译添加到您的项目的两种替代方法

您仍然可以使用 CUDA_COMPILE_PTX但是通过使用 add_custom_command 重命名来解决这个问题:add_custom_command(OUTPUT ${CMAKE_BINARY_DIR}/my_ptx.ptx COMMAND ${CMAKE_COMMAND} -E copy ${cuda_ptx_files} ${CMAKE_BINARY_DIR}/my_ptx.ptx DEPENDS ${cuda_ptx_files})并使用 ${CMAKE_BINARY_DIR}/my_ptx.ptx在 add_custom_target

使用 CMake 编译 CUDA PTX 有更好的选择。一个很好的宏示例:https://github.com/nvpro-samples/shared_sources/blob/master/cmake/private/FindCuda.cmake .可以在此处找到如何使用此宏的示例:https://github.com/nvpro-samples/gl_cuda_interop_pingpong_st/blob/master/CMakeLists.txt .