mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-04-20 06:49:15 +00:00
This reverts commit 2cbbf5dcb3.
This commit is contained in:
@@ -175,9 +175,9 @@ struct ImageToColumn
|
||||
{
|
||||
const auto [M, K] = CalculateMKDims(kargs);
|
||||
|
||||
const index_t iM = amd_wave_read_first_lane(blockIdx.x * kMPerBlock);
|
||||
const index_t iK = amd_wave_read_first_lane(blockIdx.y * kKPerBlock);
|
||||
const index_t iBatch = amd_wave_read_first_lane(blockIdx.z);
|
||||
const index_t iM = __builtin_amdgcn_readfirstlane(blockIdx.x * kMPerBlock);
|
||||
const index_t iK = __builtin_amdgcn_readfirstlane(blockIdx.y * kKPerBlock);
|
||||
const index_t iBatch = __builtin_amdgcn_readfirstlane(blockIdx.z);
|
||||
|
||||
const auto in_offset = iBatch * kargs.image_g_n_c_wis_strides[I0];
|
||||
const auto out_offset = iBatch * kargs.gemm_g_m_k_strides[I0];
|
||||
|
||||
Reference in New Issue
Block a user