ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-03 13:48:30 +00:00
Code Issues Packages Projects Releases Wiki Activity

better doc

Browse Source
This commit is contained in:
Tianxing Wu
2025-09-11 16:54:45 +00:00
parent 797031beea
commit e650db535b
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
include/ck_tile/ops/gemm/kernel/gemm_tile_partitioner.hpp
View File

@@ -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;
}