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

Fix async acc

Browse Source
This commit is contained in:
Ding, Yi
2026-01-29 10:17:36 +00:00
parent bfd9d2382a
commit d92e8010f1
1 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
include/ck_tile/core/tensor/tile_window.hpp
View File

@@ -649,12 +649,19 @@ struct tile_window_with_static_distribution
return 0;
}();
this->get_bottom_tensor_view().template async_get_vectorized_elements<vector_t>(
smem,
bottom_tensor_thread_coord.get_offset() + offset,
dram_ys_offset - imm_valid,
number<imm_valid>{},
bool_constant<oob_conditional_check>{});
if constexpr(!static_move_ys)
this->get_bottom_tensor_view().template async_get_vectorized_elements<vector_t>(
smem,
bottom_tensor_thread_coord,
offset + dram_ys_offset,
bool_constant<oob_conditional_check>{});
else
this->get_bottom_tensor_view().template async_get_vectorized_elements<vector_t>(
smem,
bottom_tensor_thread_coord.get_offset() + offset,
dram_ys_offset - imm_valid,
number<imm_valid>{},
bool_constant<oob_conditional_check>{});
// Move thread coordinate if not last access
if constexpr(iCoordAccess != (NumAccessPerCoord - 1))