From ff6f33d4f73fd5fa6b0e76d4a283fbed908673c6 Mon Sep 17 00:00:00 2001 From: danyao12 Date: Thu, 23 May 2024 15:18:43 +0800 Subject: [PATCH] add bwd validation stream_config --- example/ck_tile/01_fmha/fmha_bwd.cpp | 3 ++- example/ck_tile/01_fmha/script/smoke_test_bwd.sh | 2 +- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/example/ck_tile/01_fmha/fmha_bwd.cpp b/example/ck_tile/01_fmha/fmha_bwd.cpp index 14c33f2c58..01616650c3 100644 --- a/example/ck_tile/01_fmha/fmha_bwd.cpp +++ b/example/ck_tile/01_fmha/fmha_bwd.cpp @@ -734,7 +734,8 @@ bool run(const ck_tile::ArgParser& arg_parser) dq_buf.SetZero(); dbias_buf.SetZero(); - fmha_bwd(fmha_traits, fmha_args, stream_config); + ck_tile::stream_config stream_config_v{nullptr, true, /* log_level = */ (kname ? 1 : 0), 0, 1}; + fmha_bwd(fmha_traits, fmha_args, stream_config_v); dq_buf.FromDevice(dq_host.data()); dk_buf.FromDevice(dk_host.data()); diff --git a/example/ck_tile/01_fmha/script/smoke_test_bwd.sh b/example/ck_tile/01_fmha/script/smoke_test_bwd.sh index e4a38dfce8..9fe795471b 100644 --- a/example/ck_tile/01_fmha/script/smoke_test_bwd.sh +++ b/example/ck_tile/01_fmha/script/smoke_test_bwd.sh @@ -7,7 +7,7 @@ KNAME=1 export CK_WARMUP=0 export CK_REPEAT=1 -COMMON_ARGS='-v=1 -warmup=0 -repeat=1' +COMMON_ARGS='-v=1' for prec in "fp16" "bf16" ; do for perm in 0 1 ; do