Commit 684e012e authored by Po-Yen, Chen's avatar Po-Yen, Chen
Browse files

Use more stricter error threshold

parent 73c3a98f
...@@ -77,7 +77,7 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem) ...@@ -77,7 +77,7 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem)
} }
return ck::utils::check_err( return ck::utils::check_err(
b.mData, host_b.mData, "Error: incorrect results in output tensor", 1e-10, 1e-10); b.mData, host_b.mData, "Error: incorrect results in output tensor", 1e-6, 1e-6);
#else #else
// extend tensor shape from [N, H, W] to [N, H, W, NUM_ELEMS_IN_BUNDLE] // extend tensor shape from [N, H, W] to [N, H, W, NUM_ELEMS_IN_BUNDLE]
using DataType = detail::get_bundled_t<ADataType, NUM_ELEMS_IN_BUNDLE>; using DataType = detail::get_bundled_t<ADataType, NUM_ELEMS_IN_BUNDLE>;
...@@ -104,8 +104,8 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem) ...@@ -104,8 +104,8 @@ bool run_permute(const ExecutionConfig& config, const Problem& problem)
b.mDesc.GetElementSpaceSize() * NUM_ELEMS_IN_BUNDLE}, b.mDesc.GetElementSpaceSize() * NUM_ELEMS_IN_BUNDLE},
ck::span<const DataType>{extended_host_b.mData}, ck::span<const DataType>{extended_host_b.mData},
"Error: incorrect results in output tensor", "Error: incorrect results in output tensor",
1e-5, 1e-6,
1e-5); 1e-6);
#endif #endif
} }
......
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