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 int4_t support for check_err()

Browse Source
This commit is contained in:
Po-Yen, Chen
2022-08-19 00:30:28 -04:00
parent 194faf7837
commit 30ed3e218c
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -149,10 +149,11 @@ check_err(const std::vector<T>& out,
return res;
}
template <typename T>
typename std::enable_if<std::is_integral<T>::value && !std::is_same<T, bhalf_t>::value, bool>::type
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<T>& ref,
const std::vector<U>& ref,
const std::string& msg = "Error: Incorrect results!",
double = 0,
double atol = 0)