From 30ed3e218cbdc6e8a0a9dffc74527802369251e7 Mon Sep 17 00:00:00 2001
From: "Po-Yen, Chen" <PoYen.Chen@amd.com>
Date: Fri, 19 Aug 2022 00:30:28 -0400
Subject: [PATCH] Add int4_t support for check_err()

---
 library/include/ck/library/utility/check_err.hpp | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/library/include/ck/library/utility/check_err.hpp b/library/include/ck/library/utility/check_err.hpp
index de09ed873d..aaab32b8ad 100644
--- a/library/include/ck/library/utility/check_err.hpp
+++ b/library/include/ck/library/utility/check_err.hpp
@@ -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)