ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-06 05:55:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add type constraints for integer version check_err<>()

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-08-19 03:48:20 -04:00
parent f2c148efae
commit 726c115393
1 changed files with 9 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
library/include/ck/library/utility/check_err.hpp
View File

@@ -15,6 +15,7 @@
#include "ck/ck.hpp"
#include "ck/utility/data_type.hpp"
#include "ck/utility/type.hpp"
#include "ck/host_utility/io.hpp"
namespace ck {
@@ -149,14 +150,14 @@ check_err(const std::vector<T>& out,
return res;
}
template <typename T, typename U>
std::enable_if_t<
((std::is_signed_v<T> && std::is_integral_v<T>) || std::is_same_v<T, ck::int4_t>)&&(
std::is_signed_v<U>&& std::is_integral_v<U>)&&(sizeof(T) <= sizeof(U)) &&
!std::is_same_v<T, bhalf_t>,
bool>
check_err(const std::vector<T>& out,
const std::vector<U>& ref,
template <typename Out, typename Ref>
std::enable_if_t<((is_signed_integral_v<Out> ||
std::is_same_v<Out, ck::int4_t>)&&is_signed_integral_v<Ref>)&&(sizeof(Out) <=
sizeof(Ref)) &&
(sizeof(Ref) <= sizeof(int64_t)) && !std::is_same_v<Out, bhalf_t>,
bool>
check_err(const std::vector<Out>& out,
const std::vector<Ref>& ref,
const std::string& msg = "Error: Incorrect results!",
double = 0,
double atol = 0)