From 3893fb78d9fb2907e174e4135e23a02e1a25e5c2 Mon Sep 17 00:00:00 2001 From: Veera Rajasekhar Reddy Gopu Date: Thu, 29 Jan 2026 17:22:59 +0000 Subject: [PATCH 1/3] [Hotfix] fixed gpu arch args in aiter build from source --- transformer_engine/common/ck_fused_attn/CMakeLists.txt | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/transformer_engine/common/ck_fused_attn/CMakeLists.txt b/transformer_engine/common/ck_fused_attn/CMakeLists.txt index bc4b5d598..5e68863d5 100644 --- a/transformer_engine/common/ck_fused_attn/CMakeLists.txt +++ b/transformer_engine/common/ck_fused_attn/CMakeLists.txt @@ -30,7 +30,6 @@ message(STATUS "AITER V3_ASM_ARCHS: ${V3_ASM_ARCHS}") # pass GPU_TARGET as env into aiter codegen list(JOIN V3_ASM_ARCHS ";" V3_ASM_ARCHS_STR) -set(ENV{GPU_ARCHS} "${V3_ASM_ARCHS_STR}") if(DEFINED AITER_MHA_PATH) message(STATUS "[AITER-PREBUILT] Using AITER_MHA_PATH=${AITER_MHA_PATH}") @@ -55,7 +54,7 @@ else() COMMAND bash ${CMAKE_CURRENT_LIST_DIR}/aiter_build.sh --aiter-dir ${__AITER_SOURCE_DIR} --aiter-test-dir ${__AITER_TEST_DIR} - --gpu-archs ${V3_ASM_ARCHS_STR} + --gpu-archs "${V3_ASM_ARCHS_STR}" --ck-tile-bf16 ${CK_FUSED_ATTN_FLOAT_TO_BFLOAT16_DEFAULT} ) # libmha_fwd.so and libmha_bwd.so will be under 3rdparty/aiter/op_tests/cpp/mha From c94acf62671fbd796afaf502761c532c123bc391 Mon Sep 17 00:00:00 2001 From: Veera Rajasekhar Reddy Gopu Date: Thu, 29 Jan 2026 17:25:21 +0000 Subject: [PATCH 2/3] Fixed spaces --- transformer_engine/common/ck_fused_attn/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformer_engine/common/ck_fused_attn/CMakeLists.txt b/transformer_engine/common/ck_fused_attn/CMakeLists.txt index 5e68863d5..b6d6c7bf3 100644 --- a/transformer_engine/common/ck_fused_attn/CMakeLists.txt +++ b/transformer_engine/common/ck_fused_attn/CMakeLists.txt @@ -54,7 +54,7 @@ else() COMMAND bash ${CMAKE_CURRENT_LIST_DIR}/aiter_build.sh --aiter-dir ${__AITER_SOURCE_DIR} --aiter-test-dir ${__AITER_TEST_DIR} - --gpu-archs "${V3_ASM_ARCHS_STR}" + --gpu-archs "${V3_ASM_ARCHS_STR}" --ck-tile-bf16 ${CK_FUSED_ATTN_FLOAT_TO_BFLOAT16_DEFAULT} ) # libmha_fwd.so and libmha_bwd.so will be under 3rdparty/aiter/op_tests/cpp/mha From 6f5ae98ca59198e6d6dff34b5e8583ebdb6a8057 Mon Sep 17 00:00:00 2001 From: Veera Rajasekhar Reddy Gopu Date: Thu, 29 Jan 2026 17:26:04 +0000 Subject: [PATCH 3/3] Fixed spaces --- transformer_engine/common/ck_fused_attn/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/transformer_engine/common/ck_fused_attn/CMakeLists.txt b/transformer_engine/common/ck_fused_attn/CMakeLists.txt index b6d6c7bf3..1c61596c0 100644 --- a/transformer_engine/common/ck_fused_attn/CMakeLists.txt +++ b/transformer_engine/common/ck_fused_attn/CMakeLists.txt @@ -54,7 +54,7 @@ else() COMMAND bash ${CMAKE_CURRENT_LIST_DIR}/aiter_build.sh --aiter-dir ${__AITER_SOURCE_DIR} --aiter-test-dir ${__AITER_TEST_DIR} - --gpu-archs "${V3_ASM_ARCHS_STR}" + --gpu-archs "${V3_ASM_ARCHS_STR}" --ck-tile-bf16 ${CK_FUSED_ATTN_FLOAT_TO_BFLOAT16_DEFAULT} ) # libmha_fwd.so and libmha_bwd.so will be under 3rdparty/aiter/op_tests/cpp/mha