use integer value for GEMM test (#219)

8a2c69ee · Chao Liu · GitHub · c77ae65d · 8a2c69ee
Unverified Commit 8a2c69ee authored Apr 30, 2022 by Chao Liu Committed by GitHub Apr 30, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 9 deletions

test/gemm/gemm_util.hpp test/gemm/gemm_util.hpp +2 -9

No files found.
--- a/test/gemm/gemm_util.hpp
+++ b/test/gemm/gemm_util.hpp
@@ -139,17 +139,10 @@ struct TestGemm
        Tensor<CDataType> c_m_n_device_result(
            f_host_tensor_descriptor(params.M, params.N, params.StrideC, CLayout{}));
-        auto f_generate_tensor_value = [](auto& desc, auto type) {
+        auto f_generate_tensor_value = [](auto& tensor, auto type) {
            using dataType = decltype(type);
-            if(std::is_same<dataType, int8_t>::value)
+            tensor.GenerateTensorValue(GeneratorTensor_2<dataType>{-5, 5});
-            {
-                desc.GenerateTensorValue(GeneratorTensor_2<int8_t>{-5, 5});
-            }
-            else
-            {
-                desc.GenerateTensorValue(GeneratorTensor_3<dataType>{-0.5, 0.5});
-            }
        };
        f_generate_tensor_value(a_m_k, ADataType{});