Use AccDataType

3d091db2 · rocking · 6c0636c5 · 3d091db2
Commit 3d091db2 authored Jun 14, 2022 by rocking
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

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

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
@@ -181,8 +181,8 @@ void host_gemm_layernorm(Tensor<LayerNormOutDataType>& out_m_n,
    auto reduceSumOpInst = ReduceSumOp{};
    for(int m = 0; m < M; ++m)
    {
-        float mean_acc        = reduceSumOpInst.GetIdentityValue();
-        float square_mean_acc = reduceSumOpInst.GetIdentityValue();
+        AccDataType mean_acc        = reduceSumOpInst.GetIdentityValue();
+        AccDataType square_mean_acc = reduceSumOpInst.GetIdentityValue();

        for(int n = 0; n < N; ++n)
        {