mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-07-12 18:17:58 +00:00
fix gfx11
This commit is contained in:
@@ -203,9 +203,9 @@ auto shuffle_b(const ck_tile::HostTensor<T>& t)
|
||||
int divisor = 1;
|
||||
ck_tile::HostTensor<T> t_view({n_ / GemmConfig::N_Warp_Tile,
|
||||
GemmConfig::N_Warp_Tile,
|
||||
k_ / ItemsPerAccess,
|
||||
k_ / GemmConfig::K_Warp_Tile,
|
||||
divisor,
|
||||
ItemsPerAccess});
|
||||
GemmConfig::K_Warp_Tile / divisor});
|
||||
std::copy(t.begin(), t.end(), t_view.begin());
|
||||
return ck_tile::reference_permute(t_view, {0, 2, 3, 1, 4});
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user