Skip to content

GitLab

Commit 726320d9 authored by Chao Liu's avatar Chao Liu
Browse files

update example

parent fd444bef
Show whitespace changes
Inline Side-by-side
Showing with 6 additions and 2 deletions
example/35_splitK_gemm/run_splitK_gemm_example.inc
View file @ 726320d9
...@@ -135,14 +135,18 @@ bool run_splitK_gemm(const ProblemSize& problem_size, const ExecutionConfig& con ...@@ -135,14 +135,18 @@ bool run_splitK_gemm(const ProblemSize& problem_size, const ExecutionConfig& con
ref_invoker.Run(ref_argument); ref_invoker.Run(ref_argument);
if(std::is_same<CDataType, ck::half_t>::value) if(std::is_same<CDataType, ck::half_t>::value)
{
return ck::utils::check_err(c_m_n_device_result.mData, return ck::utils::check_err(c_m_n_device_result.mData,
c_m_n_host_result.mData, c_m_n_host_result.mData,
"fp16 incorrect result", "fp16 incorrect result",
1e-5, 3e-3,
config.init_method == 1 ? 16 : 0.1); 1e-3);
}
else else
{
return ck::utils::check_err(c_m_n_device_result.mData, c_m_n_host_result.mData); return ck::utils::check_err(c_m_n_device_result.mData, c_m_n_host_result.mData);
} }
}
return true; return true;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment