fix

Tencent · Jun 24, 2023 · a68289a · a68289a
1 parent bd7a4f2
commit a68289a
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/src/layer/loongarch/unaryop_loongarch.cpp b/src/layer/loongarch/unaryop_loongarch.cpp
@@ -109,7 +109,7 @@ struct unary_op_floor
 #if __loongarch_sx
     __m128 func_pack4(const __m128& x) const
     {
-        return __lsx_vffint_s_w(__lsx_vfrintrm_s(x));
+        return (__m128)__lsx_vfrintrm_s(x);
     }
 #endif // __loongarch_sx
 };
@@ -123,7 +123,7 @@ struct unary_op_ceil
 #if __loongarch_sx
     __m128 func_pack4(const __m128& x) const
     {
-        return __lsx_vffint_s_w(__lsx_vfrintrp_s(x));
+        return (__m128)__lsx_vfrintrp_s(x);
     }
 #endif // __loongarch_sx
 };
@@ -391,9 +391,9 @@ struct unary_op_round
 #endif
     }
 #if __loongarch_sx
-    v4f32 func_pack4(const v4f32& x) const
+    __m128 func_pack4(const __m128& x) const
     {
-        return __lsx_vffint_s_w(__lsx_vfrintrne_s(x));
+        return (__m128)__lsx_vfrintrne_s(x);
     }
 #endif // __loongarch_sx
 };
@@ -405,9 +405,9 @@ struct unary_op_trunc
         return (float)truncf(x);
     }
 #if __loongarch_sx
-    v4f32 func_pack4(const v4f32& x) const
+    __m128 func_pack4(const __m128& x) const
     {
-        return __lsx_vffint_s_w(__lsx_vftintrz_w_s(x));
+        return (__m128)__lsx_vftintrz_w_s(x);
     }
 #endif // __loongarch_sx
 };