From b2db644dcd0114ad91a1ae20c0b74beed7898e21 Mon Sep 17 00:00:00 2001 From: Qianfeng Zhang Date: Fri, 6 Jun 2025 14:22:07 +0000 Subject: [PATCH] Add assert(contextual_seqlen >= 0) in example --- example/ck_tile/18_hstu_attention/example_hstu_attention.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp index 4c17a91688..f50f7d8d88 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp @@ -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;