Refine type

f9d22b02 · rocking · 94d5f723 · f9d22b02
Commit f9d22b02 authored Jun 14, 2022 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

example/21_gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp ..._gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp +3 -3

No files found.
--- a/example/21_gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp
+++ b/example/21_gemm_layernorm/gemm_bias_relu_add_layernorm_xdl_fp16.cpp
@@ -206,9 +206,9 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n,
    {
        for(int n = 0; n < N; ++n)
        {
-            float out_f32 = 0;
+            AccDataType out_acc = 0;
-            layerNormInst(out_f32, c_m_n(m, n), mean_m(m), meanSquare_m(m), gamma_n(n), beta_n(n));
+            layerNormInst(out_acc, c_m_n(m, n), mean_m(m), meanSquare_m(m), gamma_n(n), beta_n(n));
-            out_m_n(m, n) = static_cast<DDataType>(out_f32);
+            out_m_n(m, n) = static_cast<DDataType>(out_acc);
        }
    }
 }