OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Use CUDA language in generated CMakeLists.txt from cpp_builder.py (#155979)

Browse Source 
The CMake CUDA module has been deprecated.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/155979
Approved by: https://github.com/ezyang
This commit is contained in:
cyy 2025-06-14 06:52:51 +00:00 committed by PyTorch MergeBot
parent c843909d9e
commit 1393f71e07
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
torch/_inductor/cpp_builder.py
View File

@ -1757,7 +1757,8 @@ class CppBuilder:
current_arch = _nvcc_arch_as_compile_option()
contents += textwrap.dedent(
f"""
find_package(CUDA REQUIRED)
enable_language(CUDA)
find_package(CUDAToolkit REQUIRED)
find_program(OBJCOPY_EXECUTABLE objcopy)
if(NOT OBJCOPY_EXECUTABLE)
@ -1785,7 +1786,7 @@ class CppBuilder:
# --- PTX to FATBIN Command & Target ---
add_custom_command(
OUTPUT ${{FATBIN_FILE}}
COMMAND ${{CUDA_NVCC_EXECUTABLE}} --fatbin ${{PTX_FILE}} -o ${{FATBIN_FILE}} ${{NVCC_GENCODE_FLAGS}}
COMMAND ${{CUDAToolkit_NVCC_EXECUTABLE}} --fatbin ${{PTX_FILE}} -o ${{FATBIN_FILE}} ${{NVCC_GENCODE_FLAGS}}
-gencode arch=compute_80,code=compute_80
-gencode arch=compute_{current_arch},code=sm_{current_arch}
DEPENDS ${{PTX_FILE}}