ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-11 17:51:40 +00:00
Code Issues Packages Projects Releases Wiki Activity

flag mask class interface IsTokenPairInsideMask() as const

Browse Source
This commit is contained in:
Qianfeng Zhang
2026-07-07 11:13:02 +00:00
parent 4948520212
commit db5cd5ed2d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
example/ck_tile/18_hstu_attention/hstu_block_masking.hpp
View File

@@ -186,7 +186,7 @@ struct HstuCrossAttentionBlockMaskWithLocal
};
}
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col)
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col) const
{
int row_id;
int col_id;
@@ -425,7 +425,7 @@ struct HstuSelfAttentionBlockMaskWithLocal
};
}
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col)
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col) const
{
int row_id;
int col_id;
@@ -577,7 +577,7 @@ struct HstuCrossAttentionBlockMaskNoLocal
};
}
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col)
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col) const
{
int row_id;
int col_id;
@@ -713,7 +713,7 @@ struct HstuSelfAttentionBlockMaskNoLocal
};
}
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col)
CK_TILE_HOST_DEVICE bool IsTokenPairInsideMask(int row, int col) const
{
int row_id;
int col_id;