From e650db535bac6f58eaada81078643a28de7486ea Mon Sep 17 00:00:00 2001
From: Tianxing Wu <chi0chu319@gmail.com>
Date: Thu, 11 Sep 2025 16:54:45 +0000
Subject: [PATCH] better doc

---
 .../ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp    | 9 +++++++++
 1 file changed, 9 insertions(+)

diff --git a/include/ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp b/include/ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp
index 8dceacdcc8..381edd18c8 100644
--- a/include/ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp
+++ b/include/ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp
@@ -297,6 +297,15 @@ struct GemmSpatiallyLocalTilePartitioner
                 + local_id;
         }
 
+        /**
+         * original ids: [0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15]
+         * XCD 0 gets: [0, 8], XCD 1 gets: [1, 9], ...
+         * 
+         * post-remap ids: [0, 2, 4, 6, 8, 10, 12, 14, 1, 3, 5, 7, 9, 11, 13, 15]
+         * XCD 0 gets: [0, 1], XCD 1 gets: [2, 3], ...
+         * 
+         * after remap the ids are continguous on each XCD
+         */
         return block_1d_id;
     }