add check_integer_err

example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_train_v1.cpp

@@ -1396,7 +1396,7 @@ int run(int argc, char* argv[])
         }
 
         std::cout << "Checking z:\n";
-        pass &= ck::utils::check_err(z_fwd_gs_ms_ns.mData, z_bwd_gs_ms_ns.mData, 1);
+        pass &= ck::utils::check_integer_err(z_fwd_gs_ms_ns.mData, z_bwd_gs_ms_ns.mData, 1);
 
         std::cout << "Checking y:\n";
         pass &= ck::utils::check_err(

example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_train_v2.cpp

@@ -969,7 +969,7 @@ int run(int argc, char* argv[])
         }
 
         std::cout << "Checking z:\n";
-        pass &= ck::utils::check_err(z_fwd_gs_ms_ns.mData, z_bwd_gs_ms_ns.mData, 1);
+        pass &= ck::utils::check_integer_err(z_fwd_gs_ms_ns.mData, z_bwd_gs_ms_ns.mData, 1);
 
         std::cout << "Checking y:\n";
         pass &= ck::utils::check_err(

example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_train_v1.cpp

@@ -1420,7 +1420,7 @@ int run(int argc, char* argv[])
             }
 
             std::cout << "Checking z:\n";
-            pass &= ck::utils::check_err(z_fwd_tensors[i].mData, z_bwd_tensors[i].mData, 1);
+            pass &= ck::utils::check_integer_err(z_fwd_tensors[i].mData, z_bwd_tensors[i].mData, 1);
 
             std::cout << "Checking y:\n";
             pass &= ck::utils::check_err(

example/32_batched_gemm_scale_softmax_gemm/grouped_multihead_attention_train_v2.cpp

@@ -994,7 +994,7 @@ int run(int argc, char* argv[])
             }
 
             std::cout << "Checking z:\n";
-            pass &= ck::utils::check_err(z_fwd_tensors[i].mData, z_bwd_tensors[i].mData, 1);
+            pass &= ck::utils::check_integer_err(z_fwd_tensors[i].mData, z_bwd_tensors[i].mData, 1);
 
             std::cout << "Checking y:\n";
             pass &= ck::utils::check_err(

library/include/ck/library/utility/check_err.hpp

@@ -219,7 +219,7 @@ typename std::enable_if<
     std::is_same_v<ranges::range_value_t<Range>, ranges::range_value_t<RefRange>> &&
         std::is_same_v<ranges::range_value_t<Range>, unsigned short>,
     bool>::type
-check_err(const Range& out, const RefRange& ref, unsigned short atol = 1)
+check_integer_err(const Range& out, const RefRange& ref, unsigned short atol)
 {
     const std::string& msg = "Error: Incorrect U16 results!";
     if(out.size() != ref.size())
@@ -262,7 +262,7 @@ typename std::enable_if<
     std::is_same_v<ranges::range_value_t<Range>, ranges::range_value_t<RefRange>> &&
         std::is_same_v<ranges::range_value_t<Range>, int32_t>,
     bool>::type
-check_err(const Range& out, const RefRange& ref, int32_t atol = 1)
+check_integer_err(const Range& out, const RefRange& ref, int32_t atol)
 {
     const std::string& msg = "Error: Incorrect U16 results!";
     if(out.size() != ref.size())