ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-12 02:05:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

debugging

Browse Source
This commit is contained in:
khuagarw
2026-01-13 18:26:37 +00:00
parent e69370d29d
commit cbe8d381cc
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck_tile/host/tensor_shuffle_utils.hpp
View File

@@ -158,10 +158,12 @@ auto shuffle_b_permuteN(const ck_tile::HostTensor<T>& t, const GemmConfig& gemmC
int k_ = t.get_lengths()[0];
printf("KxN: %dx%d\n", k_, n_);
int NRepeat = gemmConfig.N_Tile / gemmConfig.N_Warp_Tile / gemmConfig.N_Warp;
printf("gemmConfig.N_tile: %d, gemmConfig.N_warp_tile: %d, gemmConfig.N_Warp: %d \n",
printf("gemmConfig.N_tile: %d, gemmConfig.N_warp_tile: %d, gemmConfig.N_Warp: %d, "
"gemmConfig.K_warp_tile: %d\n",
gemmConfig.N_Tile,
gemmConfig.N_Warp_Tile,
gemmConfig.N_Warp);
gemmConfig.N_Warp,
gemmConfig.K_Warp_Tile);
if(ck_tile::is_gfx12_supported())
{
constexpr int divisor = 2;