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

Allow unsigned integer arguments for check_err()

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-08-19 15:19:05 -04:00
parent 3b0f97f6eb
commit 47770c857b
1 changed files with 3 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -151,14 +151,11 @@ check_err(const std::vector<T>& out,
}
template <typename T>
std::enable_if_t<
std::enable_if_t<(std::is_integral_v<T> && !std::is_same_v<T, bhalf_t>)
#ifdef CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4
(is_signed_integral_v<T> || std::is_same_v<T, ck::int4_t>)&&
#else
is_signed_integral_v<T> &&
|| std::is_same_v<T, int4_t>
#endif
sizeof(T) <= sizeof(int64_t),
bool>
bool>
check_err(const std::vector<T>& out,
const std::vector<T>& ref,
const std::string& msg = "Error: Incorrect results!",