ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-15 10:37:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

Merge commit '191c62967bf05f58641725b88f038bea462fe651' into develop

Browse Source
This commit is contained in:
assistant-librarian[bot]
2025-08-11 13:24:17 +00:00
parent 63c6b9b93c
commit b2ec5dde0a
4 changed files with 7 additions and 63 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/ops/reduce/kernel/reduce2d_kernel.hpp
View File

@@ -189,7 +189,9 @@ struct Reduce
/// @note Requirements:
/// - y_continous_dim % ThreadTile_N == 0 (for proper thread distribution)
/// - input_strides[-1] == 1 (for contiguous memory access)
CK_TILE_HOST static bool IsSupportedArgument(index_t y_continous_dim, auto input_strides)
template <typename InputStrides>
CK_TILE_HOST static bool IsSupportedArgument(index_t y_continous_dim,
InputStrides input_strides)
{
using S = typename Problem::BlockShape;