Revert "run clang-format==12"

This reverts commit aa59ecaa.

Revert "run clang-format==12"
This reverts commit aa59ecaa.
2c5a7323 · Po Yen Chen · GitHub · 2037a915 · 2c5a7323
Commit 2c5a7323 authored Jan 09, 2025 by Po Yen Chen Committed by GitHub Jan 09, 2025
Show whitespace changes
Inline Side-by-side

Showing with 29 additions and 15 deletions

include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp ...k_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp +29 -15

No files found.
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_pipeline_qs_ks_vs.hpp
@@ -127,25 +127,39 @@ struct BlockFmhaPipelineQSKSVS
              typename OAccElementFunction,
              typename PositionEncoding>
    CK_TILE_HOST_DEVICE auto
-    operator()(const QDramBlockWindowTmp& q_dram_block_window_tmp, // M0*K0 tile
+    operator()(const QDramBlockWindowTmp & q_dram_block_window_tmp, // M0*K0 tile
-               const QElementFunction& q_element_func,
+               const QElementFunction &
-               const KDramBlockWindowTmp& k_dram_block_window_tmp, // N0*K0 tile
+                   q_element_func,
-               const KElementFunction& k_element_func,
+               const KDramBlockWindowTmp &
-               const VDramBlockWindowTmp& v_dram_block_window_tmp, // N1*K1 tile
+                   k_dram_block_window_tmp, // N0*K0 tile
-               const VElementFunction& v_element_func,
+               const KElementFunction &
-               const BiasDramBlockWindowTmp& bias_dram_block_window_tmp, // M0*N0 tile
+                   k_element_func,
-               const BiasElementFunction& bias_element_func,
+               const VDramBlockWindowTmp &
-               RandValDramBlockWindowTmp& randval_dram_block_window_tmp,
+                   v_dram_block_window_tmp, // N1*K1 tile
-               LSEDramBlockWindowTmp& lse_dram_window_tmp, // M0*1 tile
+               const VElementFunction &
-               const LSEElementFunction& lse_element_func,
+                   v_element_func,
-               const SAccElementFunction& s_acc_element_func,
+               const BiasDramBlockWindowTmp &
-               const PComputeElementFunction& p_compute_element_func,
+                   bias_dram_block_window_tmp, // M0*N0 tile
-               const OAccElementFunction& o_acc_element_func,
+               const BiasElementFunction &
+                   bias_element_func,
+               RandValDramBlockWindowTmp &
+                   randval_dram_block_window_tmp,
+               LSEDramBlockWindowTmp &
+                   lse_dram_window_tmp, // M0*1 tile
+               const LSEElementFunction &
+                   lse_element_func,
+               const SAccElementFunction &
+                   s_acc_element_func,
+               const PComputeElementFunction &
+                   p_compute_element_func,
+               const OAccElementFunction &
+                   o_acc_element_func,
               FmhaMask mask,
               PositionEncoding position_encoding,
               float scale_s,
               void* smem_ptr,
-               DropoutType& dropout) const
+               DropoutType &
+                   dropout) const
    {
        static_assert(
            std::is_same_v<QDataType, remove_cvref_t<typename QDramBlockWindowTmp::DataType>> &&