diff --git a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp index 996b651041..6dc5bb8443 100644 --- a/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp +++ b/example/ck_tile/18_hstu_attention/example_hstu_attention.cpp @@ -163,8 +163,8 @@ static void show_hstu_attention_fwd_param(std::ostream& os, HstuAttentionFwdPara template auto get_elimit() { - double rtol = 1e-2; - double atol = 1e-2; + double rtol = 1e-3; + double atol = 1e-3; return ck_tile::make_tuple(rtol, atol); } @@ -172,8 +172,8 @@ auto get_elimit() template <> auto get_elimit() { - double rtol = 2e-2; - double atol = 2e-2; + double rtol = 2e-3; + double atol = 2e-3; return ck_tile::make_tuple(rtol, atol); } @@ -475,7 +475,8 @@ bool run(const ck_tile::ArgParser& arg_parser) // dumpBufferToFile("output_host.dat", o_host_ref.data(), o_host.get_element_space_size()); if(save_mask) - dumpBufferToFile("ck_hstu_mask.dat", mask_host.data(), mask_host.get_element_space_size()); + dumpBufferToFile( + "ck_hstu_mask.dat", mask_host.data(), mask_host.get_element_space_size()); auto [rtol, atol] = get_elimit();