use type_convert

53fd4f42 · Chao Liu · d7206533 · 53fd4f42
Commit 53fd4f42 authored Jun 18, 2022 by Chao Liu
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

library/include/ck/library/reference_tensor_operation/cpu/reference_gemm_bias_2d.hpp ...reference_tensor_operation/cpu/reference_gemm_bias_2d.hpp +2 -2

No files found.
--- a/library/include/ck/library/reference_tensor_operation/cpu/reference_gemm_bias_2d.hpp
+++ b/library/include/ck/library/reference_tensor_operation/cpu/reference_gemm_bias_2d.hpp
@@ -66,8 +66,8 @@ struct ReferenceGemmBias2D : public device::BaseOperator
                for(int k = 0; k < K; ++k)
                {
-                    arg.a_element_op_(a, static_cast<AccDataType>(arg.a_m_k_(m, k)));
+                    arg.a_element_op_(a, ck::type_convert<AccDataType>(arg.a_m_k_(m, k)));
-                    arg.b_element_op_(b, static_cast<AccDataType>(arg.b_k_n_(k, n)));
+                    arg.b_element_op_(b, ck::type_convert<AccDataType>(arg.b_k_n_(k, n)));
                    acc += a * b;
                }