diff --git a/example/ck_tile/01_fmha/fmha_fwd.cpp b/example/ck_tile/01_fmha/fmha_fwd.cpp index 5970aa13cb..f0c282cbb9 100644 --- a/example/ck_tile/01_fmha/fmha_fwd.cpp +++ b/example/ck_tile/01_fmha/fmha_fwd.cpp @@ -103,6 +103,23 @@ auto get_elimit(int init_method) } } +template <> +auto get_elimit(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 bool run(const ck_tile::ArgParser& arg_parser) {