ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-13 17:55:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add IsOutOfSinkBound alias in GenericAttentionMask (API compatibility)

Browse Source 
Made-with: Cursor
This commit is contained in:
root
2026-04-23 08:17:34 +00:00
parent 8506db8761
commit 3f076a6fc1
1 changed files with 6 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
include/ck_tile/ops/unified_attention/block/block_masking.hpp
View File

@@ -180,6 +180,12 @@ struct GenericAttentionMask
}
}
// Alias kept for compatibility with reference_batched_masking.hpp
CK_TILE_HOST_DEVICE constexpr auto IsOutOfSinkBound(index_t i_y, index_t i_x) const
{
return IsOutOfBound(i_y, i_x);
}
// per-pixel check if out-of-bound, if true, need mask a value(like -INF)
CK_TILE_HOST_DEVICE constexpr auto IsOutOfBound(index_t i_y, index_t i_x) const
{