提交 979d2e0b 编写于 作者: K Kexin Zhao

small fix

上级 4f1aa5bc
...@@ -426,8 +426,8 @@ __device__ inline bool operator>=(const float16& a, const float16& b) { ...@@ -426,8 +426,8 @@ __device__ inline bool operator>=(const float16& a, const float16& b) {
} }
// On ARMv8.2-A CPU // On ARMv8.2-A CPU
#elif (PADDLE_GNUC_VER >= 71 || PADDLE_CLANG_VER >= 39) && \ #elif defined(PADDLE_NEON_64) && defined(PADDLE_ARM_FP16) && \
defined(PADDLE_NEON_64) && defined(PADDLE_ARM_FP16) (PADDLE_GNUC_VER >= 71 || PADDLE_CLANG_VER >= 39)
__host__ inline float16 operator+(const float16& a, const float16& b) { __host__ inline float16 operator+(const float16& a, const float16& b) {
return float16(vaddh_f16(float16_t(a), float16_t(b))); return float16(vaddh_f16(float16_t(a), float16_t(b)));
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册