From 072c0886d53d933a7e1d3da7e8a3a91670cf181d Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andr=C3=A9=20Lustosa?= <30708955+andre-motta@users.noreply.github.com> Date: Sun, 18 Oct 2026 08:13:17 -0400 Subject: [PATCH] [Build] Keep plain CUDA arch when its suffixed variant is also requested (#56549) Signed-off-by: Andre Lustosa Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com> Co-authored-by: Shengqi Chen --- cmake/utils.cmake | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/cmake/utils.cmake b/cmake/utils.cmake index 6ded936213..83eed340eb 100644 --- a/cmake/utils.cmake +++ b/cmake/utils.cmake @@ -425,6 +425,7 @@ function(cuda_archs_loose_intersection OUT_CUDA_ARCHS SRC_CUDA_ARCHS TGT_CUDA_AR if("${_base}" IN_LIST _TGT_CUDA_ARCHS) list(REMOVE_ITEM _SRC_CUDA_ARCHS "${_arch}") list(REMOVE_ITEM _TGT_CUDA_ARCHS "${_base}") + list(REMOVE_ITEM _TGT_CUDA_ARCHS "${_base}a" "${_base}f") list(APPEND _CUDA_ARCHS "${_arch}") endif() endif() @@ -464,8 +465,10 @@ function(cuda_archs_loose_intersection OUT_CUDA_ARCHS SRC_CUDA_ARCHS TGT_CUDA_AR string(REGEX REPLACE "[af]$" "" _base "${_arch}") if ("${_base}" IN_LIST _SRC_CUDA_ARCHS) list(REMOVE_ITEM _TGT_CUDA_ARCHS "${_arch}") - list(REMOVE_ITEM _SRC_CUDA_ARCHS "${_base}") list(APPEND _CUDA_ARCHS "${_arch}") + if (NOT "${_base}" IN_LIST _TGT_CUDA_ARCHS) + list(REMOVE_ITEM _SRC_CUDA_ARCHS "${_base}") + endif() endif() endif() endforeach()