From df352f955a2063239e15823a1f5b53032e7de2df Mon Sep 17 00:00:00 2001
From: "PoYen, Chen" <PoYen.Chen@amd.com>
Date: Tue, 23 Jul 2024 02:31:45 +0000
Subject: [PATCH] Add comment

---
 .../ops/fmha/pipeline/block_fmha_fwd_appendkv_pipeline.hpp      | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_appendkv_pipeline.hpp b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_appendkv_pipeline.hpp
index d825b81cc4..194ff3ec59 100644
--- a/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_appendkv_pipeline.hpp
+++ b/include/ck_tile/ops/fmha/pipeline/block_fmha_fwd_appendkv_pipeline.hpp
@@ -185,6 +185,8 @@ struct BlockFmhaFwdAppendKVPipeline
                                  rotary_sin_block_window_tmp.get_window_origin(),
                                  Policy::template MakeRotaryCosSinTileDistribution<Problem>());
 
+            // We assume that each thread owns contiguous elements on head dimention. And we will
+            // use the distribution to enable/disable threads in order to override knew_tile content
             if constexpr(RotaryEnum == BlockRotaryEmbeddingEnum::INTERLEAVED)
             {
                 auto rotary_cos_tile = load_tile(rotary_cos_window);