Skip to content

Address CUDA arch codes #120

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
mc-nv merged 3 commits into from
Feb 9, 2026
Merged

Address CUDA arch codes #120

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
f5e901f
Updating default architecture codes
mc-nv Feb 6, 2026
850e2e0
Update copyright year
mc-nv Feb 6, 2026
7153e90
Update logic
mc-nv Feb 6, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
47 changes: 39 additions & 8 deletions cmake/define.cuda_architectures.cmake
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
# Copyright 2025, NVIDIA CORPORATION. All rights reserved.
# Copyright 2025-2026, NVIDIA CORPORATION. All rights reserved.
#
# Redistribution and use in source and binary forms, with or without
# modification, are permitted provided that the following conditions
Expand Down Expand Up @@ -27,19 +27,50 @@
function(set_cuda_architectures_list)
# Check if CUDA_ARCH_LIST environment variable is set
if(DEFINED ENV{CUDA_ARCH_LIST})
# Parse the existing CUDA_ARCH_LIST
set(cuda_arch_input "$ENV{CUDA_ARCH_LIST}")
string(REGEX REPLACE "PTX" "" cuda_arch_input "${cuda_arch_input}")
string(REGEX REPLACE " " "-real;" cuda_arch_input "${cuda_arch_input}")
string(REGEX REPLACE "-real;\$" "" cuda_arch_input "${cuda_arch_input}")
string(REGEX REPLACE "\\." "" cuda_arch_input "${cuda_arch_input}")
# Parse CUDA_ARCH_LIST: split by spaces, skip PTX, validate each code
set(raw_input "$ENV{CUDA_ARCH_LIST}")
string(REGEX REPLACE "PTX" "" raw_input "${raw_input}")
string(REPLACE " " ";" arch_list "${raw_input}")

set(cuda_arch_result_list "")
foreach(arch IN LISTS arch_list)
string(STRIP "${arch}" arch)
if(arch STREQUAL "")
continue()
endif()
# Normalize: remove dots so 10.0 -> 100, 12.0 -> 120
string(REGEX REPLACE "\\." "" arch_num "${arch}")
if(NOT arch_num MATCHES "^[0-9]+$")
continue()
endif()
# Code >= 100 (10.x, 11.x, 12.x): use family code, no -real
if(arch_num GREATER_EQUAL 100)
math(EXPR arch_major "${arch_num} / 10")
set(arch_entry "${arch_major}0f")
Comment thread
dmitry-tokarev-nv marked this conversation as resolved.
else()
set(arch_entry "${arch_num}-real")
endif()
list(APPEND cuda_arch_result_list "${arch_entry}")
Comment thread
dmitry-tokarev-nv marked this conversation as resolved.
endforeach()
# If last element is below 100 (has -real), leave it without -real
list(LENGTH cuda_arch_result_list result_len)
if(result_len GREATER 0)
math(EXPR last_index "${result_len} - 1")
list(GET cuda_arch_result_list ${last_index} last_entry)
string(REGEX REPLACE "-real$" "" last_entry_stripped "${last_entry}")
if(NOT last_entry_stripped STREQUAL last_entry)
list(REMOVE_AT cuda_arch_result_list ${last_index})
list(APPEND cuda_arch_result_list "${last_entry_stripped}")
endif()
endif()
list(JOIN cuda_arch_result_list ";" cuda_arch_input)

set(CUDA_ARCHITECTURES "${cuda_arch_input}" PARENT_SCOPE)

message(STATUS "CUDA_ARCH_LIST found, defined CUDA_ARCHITECTURES: $ENV{CUDA_ARCH_LIST}")
else()
# Set default value if CUDA_ARCH_LIST is not present
set(CUDA_ARCHITECTURES "75-real;80-real;86-real;89-real;90-real;100-real;120" PARENT_SCOPE)
set(CUDA_ARCHITECTURES "75-real;80-real;86-real;89-real;90-real;100f;120f" PARENT_SCOPE)
message(STATUS "CUDA_ARCH_LIST not found, using default values for CUDA_ARCHITECTURES: ${CUDA_ARCHITECTURES}")
endif()
endfunction()
Expand Down
Loading