From 2499b8d40121d67dcf066ddcb16cc6f5f7bf905f Mon Sep 17 00:00:00 2001 From: Clement Lin Date: Thu, 10 Apr 2025 09:13:12 +0800 Subject: [PATCH] Fix indentation --- .../flash_attention_fwd.cpp | 31 +++++++++---------- 1 file changed, 15 insertions(+), 16 deletions(-) diff --git a/example/ck_tile/99_toy_example/03_flash_attention_fwd/flash_attention_fwd.cpp b/example/ck_tile/99_toy_example/03_flash_attention_fwd/flash_attention_fwd.cpp index 299d29b279..8ce1d6c6c7 100644 --- a/example/ck_tile/99_toy_example/03_flash_attention_fwd/flash_attention_fwd.cpp +++ b/example/ck_tile/99_toy_example/03_flash_attention_fwd/flash_attention_fwd.cpp @@ -83,21 +83,21 @@ int main(int argc, char* argv[]) switch(init_method) { - case 0: break; - case 1: - ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(q_host); - ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(k_host); - ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(v_host); - break; - case 2: - ck_tile::FillUniformDistribution{-3.f, 3.f}(q_host); - ck_tile::FillUniformDistribution{-3.f, 3.f}(k_host); - ck_tile::FillUniformDistribution{-3.f, 3.f}(v_host); - break; - default: - ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(q_host); - ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(k_host); - ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(v_host); + case 0: break; + case 1: + ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(q_host); + ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(k_host); + ck_tile::FillUniformDistributionIntegerValue{-3.f, 3.f}(v_host); + break; + case 2: + ck_tile::FillUniformDistribution{-3.f, 3.f}(q_host); + ck_tile::FillUniformDistribution{-3.f, 3.f}(k_host); + ck_tile::FillUniformDistribution{-3.f, 3.f}(v_host); + break; + default: + ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(q_host); + ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(k_host); + ck_tile::FillUniformDistributionIntegerValue{-2.f, 2.f}(v_host); } ck_tile::DeviceMem q_buf(q_host.get_element_space_size_in_bytes()); ck_tile::DeviceMem k_buf(k_host.get_element_space_size_in_bytes()); @@ -127,7 +127,6 @@ int main(int argc, char* argv[]) float ave_time = ck_tile::launch_kernel(ck_tile::stream_config{nullptr, true}, ck_tile::make_kernel( - // StreamConfig{nullptr, static_cast(time_kernel)}, ck_tile::FlashAttentionFwd