From 05d021b18b07de31dfcb702eacf56e059de694de Mon Sep 17 00:00:00 2001 From: Alp Dener Date: Wed, 29 Jul 2026 17:26:00 +0000 Subject: [PATCH] adding Rubin-specific SM107a compile target under the SM100 arch family Signed-off-by: Alp Dener --- transformer_engine/common/CMakeLists.txt | 3 +++ 1 file changed, 3 insertions(+) diff --git a/transformer_engine/common/CMakeLists.txt b/transformer_engine/common/CMakeLists.txt index 13880e0d61..9734e24fbe 100644 --- a/transformer_engine/common/CMakeLists.txt +++ b/transformer_engine/common/CMakeLists.txt @@ -49,6 +49,9 @@ if(NOT arch_100_index EQUAL -1) if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 12.9) list(APPEND NVTE_SPECIFIC_ARCHS "103a") endif() + if(CUDAToolkit_VERSION VERSION_GREATER_EQUAL 13.4) + list(APPEND NVTE_SPECIFIC_ARCHS "107a") + endif() endif() # Check for architecture 101 (if we see this we are in toolkit <= 12.9)