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 3d3da4f604..a1cdbd6663 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp @@ -285,14 +285,18 @@ bool run(const ck_tile::ArgParser& arg_parser) if(is_jagged) { for(auto len : seq_lengths) - total_flops += (static_cast(len) * len * hdim_qk + len * hdim_v * len) * 2; + total_flops += + (static_cast(len) * len * hdim_qk + static_cast(len) * hdim_v * len) * + 2; total_flops *= num_head; } else { total_flops = static_cast(num_batch) * num_head * - (seqlen * seqlen * hdim_qk + seqlen * hdim_v * seqlen) * 2; + (static_cast(seqlen) * seqlen * hdim_qk + + static_cast(seqlen) * hdim_v * seqlen) * + 2; }; int batches_for_alloc = is_jagged ? 1 : num_batch; @@ -402,7 +406,8 @@ bool run(const ck_tile::ArgParser& arg_parser) params.philox_offset = 0UL; }; - show_hstu_attention_fwd_param(std::cout, params); + // show_hstu_attention_fwd_param(std::cout, params); + std::ignore = show_hstu_attention_fwd_param; hipStream_t stream;