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

Add assert(contextual_seqlen >= 0) in example

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-06-06 14:22:07 +00:00
parent d7930cd541
commit b2db644dcd
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
example/ck_tile/18_hstu_attention/example_hstu_attention.cpp
View File

@@ -296,6 +296,8 @@ bool run(const ck_tile::ArgParser& arg_parser)
assert(input_max_uih_seqlen <= 0 || input_max_uih_seqlen >= max_uih_seqlen);
assert(input_max_target <= 0 || input_max_target >= max_target);
assert(contextual_seqlen >= 0);
max_uih_seqlen = (input_max_uih_seqlen > 0) ? input_max_uih_seqlen : max_uih_seqlen;
max_target = (input_max_target > 0) ? input_max_target : max_target;