ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-13 02:27:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
Jim
2025-04-03 14:49:03 +08:00
parent 536433f5e7
commit b39e9d002b
3 changed files with 783 additions and 743 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -352,6 +352,9 @@ bool run(const ck_tile::ArgParser& arg_parser)
ck_tile::FillUniformDistribution<VDataType>{0.f, 1.f, seed}(v_host);
ck_tile::FillUniformDistribution<BiasDataType>{0.f, 1.f, seed}(bias_host);
ck_tile::FillUniformDistribution<OGradDataType>{0.f, 1.f, seed}(do_host);
// ck_tile::FillConstant<QDataType>{1}(q_host);
// ck_tile::FillConstant<KDataType>{1}(k_host);
// ck_tile::FillConstant<OGradDataType>{2}(do_host);
}
else if(init_method == 2)
{

1518
example/ck_tile/01_fmha/hsaco/bwd_bf16_swa_a32_rtna_psskddv.cpp
View File

File diff suppressed because it is too large Load Diff

5
include/ck_tile/host/check_err.hpp
View File

@@ -246,12 +246,15 @@ check_err(const Range& out,
{
max_err = err > max_err ? err : max_err;
err_count++;
if(err_count < 5)
if(err_count < 192 * 128)
{
std::cerr << msg << std::setw(12) << std::setprecision(7) << " out[" << i
<< "] != ref[" << i << "]: " << o << " != " << r << std::endl;
}
res = false;
// } else {
// std::cerr << msg << std::setw(12) << std::setprecision(7) << " out[" << i
// << "] != ref[" << i << "]: " << o << " != " << r << std::endl;
}
}
if(!res)