From e77830019cbe92efff25e5748201503bef8b9838 Mon Sep 17 00:00:00 2001
From: nihuini <nihuini@tencent.com>
Date: Tue, 3 Sep 2024 15:45:21 +0800
Subject: [PATCH] fix woa build

---
 src/layer/arm/rmsnorm_arm_asimdhp.cpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/layer/arm/rmsnorm_arm_asimdhp.cpp b/src/layer/arm/rmsnorm_arm_asimdhp.cpp
index 28f1201c136..74fcb7d7a37 100644
--- a/src/layer/arm/rmsnorm_arm_asimdhp.cpp
+++ b/src/layer/arm/rmsnorm_arm_asimdhp.cpp
@@ -50,7 +50,7 @@ static void rmsnorm_fp16s(__fp16* ptr, const float* gamma_ptr, float eps, int el
         }
         for (; i < size; i++)
         {
-            sqsum += ptr0[0] * ptr0[0];
+            sqsum += (float)ptr0[0] * (float)ptr0[0];
             ptr0++;
         }
     }
@@ -183,7 +183,7 @@ static void rmsnorm_fp16s(__fp16* ptr, const float* gamma_ptr, float eps, int el
         }
         for (; i < size; i++)
         {
-            ptr[0] = (ptr[0] * a) * gamma_ptr[0];
+            ptr[0] = (__fp16)(((float)ptr[0] * a) * gamma_ptr[0]);
             ptr++;
             gamma_ptr++;
         }
@@ -211,7 +211,7 @@ static void rmsnorm_fp16s(__fp16* ptr, const float* gamma_ptr, float eps, int el
         }
         for (; i < size; i++)
         {
-            ptr[0] = ptr[0] * a;
+            ptr[0] = (__fp16)((float)ptr[0] * a);
             ptr++;
         }
     }