From 25329086992ec6563757dae82d30406172e43e54 Mon Sep 17 00:00:00 2001 From: "PoYen, Chen" Date: Tue, 11 Jun 2024 14:34:45 +0000 Subject: [PATCH] Print num_splits conditionally --- example/ck_tile/01_fmha/fmha_fwd.cpp | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp index 28b69a7333..862d4d3e58 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -508,8 +508,12 @@ bool run(const ck_tile::ArgParser& arg_parser) : (std::string("(") + std::to_string(seqlen_kpads[0]) + ")")) << ", d:" << hdim_q << "/" << hdim_v << ", scale_s:" << scale_s << ", bias:" << bias << ", p_drop:" << p_drop << ", lse:" << lse << ", squant:" << squant - << ", mask:" << mask << ", v:" << vlayout << ", num_splits: " << num_splits - << std::flush; + << ", mask:" << mask << ", v:" << vlayout; + if(1 < num_splits) + { + std::cout << ", num_splits: " << num_splits; + } + std::cout << std::flush; auto fmha_traits = fmha_fwd_traits{hdim_q, hdim_v,