clang-format

example/ck_tile/18_flexattn/fmha_fwd.cpp

@@ -849,9 +849,9 @@ bool run(const ck_tile::ArgParser& arg_parser)
         }
         else // fmha_fwd_traits or fmha_splitkv_traits
         {
-            traits.is_group_mode       = (mode == mode_enum::group);
-            traits.mask_type = mask.type;
-            traits.bias_type = bias.type;
+            traits.is_group_mode = (mode == mode_enum::group);
+            traits.mask_type     = mask.type;
+            traits.bias_type     = bias.type;
             // traits.has_lse             = lse;
             // traits.do_fp8_static_quant = squant;
 
@@ -1504,13 +1504,13 @@ bool run(const ck_tile::ArgParser& arg_parser)
                         mask.type == mask_enum::mask_top_left));
         }
 
-        auto pre_softmax = [] (auto s) {
-            //ck_tile::detail::swallow(s);
+        auto pre_softmax = [](auto s) {
+            // ck_tile::detail::swallow(s);
             return CK_PRE_SOFTMAX_F;
         };
         s_host_ref.ForEach([&](auto& self, auto i) {
             auto new_val = pre_softmax(self(i));
-            self(i) = new_val;
+            self(i)      = new_val;
         });
 
         if(lse)

include/ck_tile/ops/fmha/kernel/fmha_flex_fwd_kernel.hpp

@@ -20,7 +20,10 @@
 
 namespace ck_tile {
 
-template <typename FmhaPipeline_, typename EpiloguePipeline_, typename ScoreModFunction_, typename PreSoftmaxFunction_>
+template <typename FmhaPipeline_,
+          typename EpiloguePipeline_,
+          typename ScoreModFunction_,
+          typename PreSoftmaxFunction_>
 struct FmhaFwdKernel
 {
     using FmhaPipeline                            = ck_tile::remove_cvref_t<FmhaPipeline_>;
@@ -1317,8 +1320,7 @@ struct FmhaFwdKernel
             };
 
         auto pre_softmax_def = PreSoftmaxFunction_{};
-        auto pre_softmax_arg = [pre_softmax_def](
-               typename PreSoftmaxFunction_::TScore s) {
+        auto pre_softmax_arg = [pre_softmax_def](typename PreSoftmaxFunction_::TScore s) {
             return pre_softmax_def(s);
         };
 

include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qr_ks_vs_async.hpp

@@ -13,7 +13,9 @@
 namespace ck_tile {
 
 // a variation of qr/ks/vs, where we use async copy to load k (potentially v in the future)
-template <typename Problem_, typename Policy_ = BlockFmhaPipelineQRKSVSAsyncDefaultPolicy, typename PreSoftmaxFunction_>
+template <typename Problem_,
+          typename Policy_ = BlockFmhaPipelineQRKSVSAsyncDefaultPolicy,
+          typename PreSoftmaxFunction_>
 struct BlockFmhaPipelineQRKSVSAsync
 {
     using Problem               = remove_cvref_t<Problem_>;