From 29cf1610f14e06ca8a652d80fd83c40bba78218c Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Tue, 20 May 2025 07:46:21 +0000 Subject: [PATCH] Enable RTN fp32 to bf16 conversion by adding compiler option in CMakeLists.txt --- example/ck_tile/18_hstu_attention/CMakeLists.txt | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/example/ck_tile/18_hstu_attention/CMakeLists.txt b/example/ck_tile/18_hstu_attention/CMakeLists.txt index 8d1bfa0d8a..e8e39afa65 100644 --- a/example/ck_tile/18_hstu_attention/CMakeLists.txt +++ b/example/ck_tile/18_hstu_attention/CMakeLists.txt @@ -10,7 +10,7 @@ target_sources(${EXAMPLE_HSTU_ATTENTION} PRIVATE ${INTERFACES_SRCS} ${INSTANCE_ set(EXAMPLE_HSTU_ATTENTION_COMPILE_OPTIONS) -list(APPEND EXAMPLE_HSTU_ATTENTION_COMPILE_OPTIONS -Wno-undefined-func-template -Wno-float-equal) +list(APPEND EXAMPLE_HSTU_ATTENTION_COMPILE_OPTIONS -Wno-undefined-func-template -Wno-float-equal -DCK_TILE_FLOAT_TO_BFLOAT16_DEFAULT=3) target_compile_options(${EXAMPLE_HSTU_ATTENTION} PRIVATE ${EXAMPLE_HSTU_ATTENTION_COMPILE_OPTIONS})