ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Set fp8 rounding error for check_err()

Browse Source
This commit is contained in:
Po Yen Chen
2024-04-08 12:32:07 +00:00
parent 641ae96215
commit e49498f616
1 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
example/ck_tile/01_fmha/fmha_fwd.cpp
View File

@@ -103,6 +103,23 @@ auto get_elimit<ck_tile::bf16_t>(int init_method)
}
}
template <>
auto get_elimit<ck_tile::fp8_t>(int init_method)
{
if(init_method == 0)
{
unsigned rounding_error = 0;
double atol = 2e-3;
return ck_tile::make_tuple(rounding_error, atol);
}
else
{
unsigned rounding_error = 1;
double atol = 0.0625;
return ck_tile::make_tuple(rounding_error, atol);
}
}
template <typename DataType>
bool run(const ck_tile::ArgParser& arg_parser)
{