Commit 38b57897 authored by rocking's avatar rocking
Browse files

Fix CI error

parent 70bb9171
...@@ -200,6 +200,8 @@ bool run_grouped_conv_fwd(bool do_verification, ...@@ -200,6 +200,8 @@ bool run_grouped_conv_fwd(bool do_verification,
std::cout << "Perf: " << avg_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s, " std::cout << "Perf: " << avg_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s, "
<< conv.GetTypeString() << std::endl; << conv.GetTypeString() << std::endl;
bool pass = true;
if(do_verification) if(do_verification)
{ {
Tensor<CShuffleDataType> c_host(out_g_n_k_wos_desc); Tensor<CShuffleDataType> c_host(out_g_n_k_wos_desc);
...@@ -232,10 +234,11 @@ bool run_grouped_conv_fwd(bool do_verification, ...@@ -232,10 +234,11 @@ bool run_grouped_conv_fwd(bool do_verification,
out_device_buf.FromDevice(out_device.mData.data()); out_device_buf.FromDevice(out_device.mData.data());
return ck::utils::check_err( pass &= ck::utils::check_err(
out_device.mData, out_host.mData, "Error: incorrect results!", 1e-5f, 1e-4f); out_device.mData, out_host.mData, "Error: incorrect results!", 1e-5f, 1e-4f);
} }
return true;
return (pass ? 0 : 1);
} }
int main() int main()
......
...@@ -184,6 +184,8 @@ bool run_grouped_conv_fwd(bool do_verification, ...@@ -184,6 +184,8 @@ bool run_grouped_conv_fwd(bool do_verification,
std::cout << "Perf: " << avg_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s, " std::cout << "Perf: " << avg_time << " ms, " << tflops << " TFlops, " << gb_per_sec << " GB/s, "
<< conv.GetTypeString() << std::endl; << conv.GetTypeString() << std::endl;
bool pass = true;
if(do_verification) if(do_verification)
{ {
auto ref_conv = ck::tensor_operation::host::ReferenceConvFwd<NDimSpatial, auto ref_conv = ck::tensor_operation::host::ReferenceConvFwd<NDimSpatial,
...@@ -210,10 +212,11 @@ bool run_grouped_conv_fwd(bool do_verification, ...@@ -210,10 +212,11 @@ bool run_grouped_conv_fwd(bool do_verification,
out_device_buf.FromDevice(out_device.mData.data()); out_device_buf.FromDevice(out_device.mData.data());
return ck::utils::check_err( pass &= ck::utils::check_err(
out_device.mData, out_host.mData, "Error: incorrect results!", 1e-5f, 1e-4f); out_device.mData, out_host.mData, "Error: incorrect results!", 1e-5f, 1e-4f);
} }
return true;
return (pass ? 0 : 1);
} }
int main() int main()
......
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