ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-07 00:04:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

add bwd validation stream_config

Browse Source
This commit is contained in:
danyao12
2024-05-23 15:18:43 +08:00
parent 826a894335
commit ff6f33d4f7
2 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
example/ck_tile/01_fmha/fmha_bwd.cpp
View File

@@ -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());

2
example/ck_tile/01_fmha/script/smoke_test_bwd.sh
View File

@@ -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