From 726c1153939228adefd31b2c52c30a84e8919e65 Mon Sep 17 00:00:00 2001
From: "Po-Yen, Chen" <PoYen.Chen@amd.com>
Date: Fri, 19 Aug 2022 03:48:20 -0400
Subject: [PATCH] Add type constraints for integer version check_err<>()

---
 .../include/ck/library/utility/check_err.hpp    | 17 +++++++++--------
 1 file changed, 9 insertions(+), 8 deletions(-)

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