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

Adjust the atol and rtol and fix the check_err() using in example_hstu_attention.cpp

Browse Source
This commit is contained in:
Qianfeng Zhang
2025-08-12 01:33:06 +00:00
parent d5d4a0d6c0
commit 30dd274e7e
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
example/ck_tile/18_hstu_attention/example_hstu_attention.cpp
View File

@@ -191,8 +191,8 @@ static void show_hstu_attention_fwd_param(std::ostream& os, HstuAttentionFwdPara
template <typename DataType>
auto get_elimit()
{
double rtol = 1e-3;
double atol = 1e-3;
double rtol = 1.6e-2;
double atol = 1e-5;
return ck_tile::make_tuple(rtol, atol);
}
@@ -200,8 +200,8 @@ auto get_elimit()
template <>
auto get_elimit<ck_tile::bf16_t>()
{
double rtol = 1e-3;
double atol = 1e-3;
double rtol = 1.6e-2;
double atol = 1e-5;
return ck_tile::make_tuple(rtol, atol);
}
@@ -545,7 +545,7 @@ bool run(const ck_tile::ArgParser& arg_parser)
auto [rtol, atol] = get_elimit<InOutDataType>();
res = ck_tile::check_err(
o_host, o_host_ref, std::string("hstu_attention output error"), atol, rtol);
o_host, o_host_ref, std::string("hstu_attention output error"), rtol, atol);
};
if(measure_perf)