Skip to content

GitLab

Commit 41b94703 authored by Andriy Roshchenko's avatar Andriy Roshchenko
Browse files

Fix GPU verification reporting logic.

parent 25c6d97b
Hide whitespace changes
Inline Side-by-side
Showing with 16 additions and 12 deletions
example/01_gemm/run_gemm_example.inc
View file @ 41b94703
......@@ -416,13 +416,15 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
c_m_n_device_ref_buf.FromDevice(c_m_n_device_ref_result.mData.data());
c_m_n_device_buf.FromDevice(c_m_n_device_result.mData.data());
pass = ck::utils::check_err(c_m_n_device_result,
c_m_n_device_ref_result,
"Error: Incorrect results!",
get_rtol<CDataType>(),
get_atol<CDataType>());
if(pass)
auto gpu_pass = ck::utils::check_err(c_m_n_device_result,
c_m_n_device_ref_result,
"Error: Incorrect results!",
get_rtol<CDataType>(),
get_atol<CDataType>());
if(gpu_pass)
std::cout << "Verification on GPU: PASS" << std::endl;
pass = pass && gpu_pass;
}
return pass;
......
example/35_splitK_gemm/run_splitK_gemm_example.inc
View file @ 41b94703
......@@ -308,13 +308,15 @@ bool run_splitK_gemm(const ProblemSize& problem_size, const ExecutionConfig& con
c_m_n_device_ref_buf.FromDevice(c_m_n_device_ref_result.mData.data());
c_m_n_device_buf.FromDevice(c_m_n_device_result.mData.data());
pass = ck::utils::check_err(c_m_n_device_result,
c_m_n_device_ref_result,
"Error: Incorrect results!",
get_rtol<CDataType>(),
get_atol<CDataType>());
if(pass)
auto gpu_pass = ck::utils::check_err(c_m_n_device_result,
c_m_n_device_ref_result,
"Error: Incorrect results!",
get_rtol<CDataType>(),
get_atol<CDataType>());
if(gpu_pass)
std::cout << "Verification on GPU: PASS" << std::endl;
pass &= gpu_pass;
}
if(config.time_kernel)
......
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