Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 182da953 编写于 作者: K Kexin Zhao
浏览文件
操作

small fix

上级 f2bbbb2b
显示空白变更内容
内联 并排
Showing with 11 addition and 0 deletion
paddle/fluid/platform/float16.h
浏览文件 @ 182da953
...@@ -490,6 +490,7 @@ HOSTDEVICE inline float16 operator+(const float16& a, const float16& b) { ...@@ -490,6 +490,7 @@ HOSTDEVICE inline float16 operator+(const float16& a, const float16& b) {
return float16(__hadd(half(a), half(b))); return float16(__hadd(half(a), half(b)));
#else #else
return float16(float(a) + float(b)); return float16(float(a) + float(b));
#endif
} }
HOSTDEVICE inline float16 operator-(const float16& a, const float16& b) { HOSTDEVICE inline float16 operator-(const float16& a, const float16& b) {
...@@ -497,6 +498,7 @@ HOSTDEVICE inline float16 operator-(const float16& a, const float16& b) { ...@@ -497,6 +498,7 @@ HOSTDEVICE inline float16 operator-(const float16& a, const float16& b) {
return float16(__hsub(half(a), half(b))); return float16(__hsub(half(a), half(b)));
#else #else
return float16(float(a) - float(b)); return float16(float(a) - float(b));
#endif
} }
HOSTDEVICE inline float16 operator*(const float16& a, const float16& b) { HOSTDEVICE inline float16 operator*(const float16& a, const float16& b) {
...@@ -504,6 +506,7 @@ HOSTDEVICE inline float16 operator*(const float16& a, const float16& b) { ...@@ -504,6 +506,7 @@ HOSTDEVICE inline float16 operator*(const float16& a, const float16& b) {
return float16(__hmul(half(a), half(b))); return float16(__hmul(half(a), half(b)));
#else #else
return float16(float(a) * float(b)); return float16(float(a) * float(b));
#endif
} }
HOSTDEVICE inline float16 operator/(const float16& a, const float16& b) { HOSTDEVICE inline float16 operator/(const float16& a, const float16& b) {
...@@ -514,6 +517,7 @@ HOSTDEVICE inline float16 operator/(const float16& a, const float16& b) { ...@@ -514,6 +517,7 @@ HOSTDEVICE inline float16 operator/(const float16& a, const float16& b) {
return float16(num / denom); return float16(num / denom);
#else #else
return float16(float(a) / float(b)); return float16(float(a) / float(b));
#endif
} }
HOSTDEVICE inline float16 operator-(const float16& a) { HOSTDEVICE inline float16 operator-(const float16& a) {
...@@ -523,6 +527,7 @@ HOSTDEVICE inline float16 operator-(const float16& a) { ...@@ -523,6 +527,7 @@ HOSTDEVICE inline float16 operator-(const float16& a) {
float16 res; float16 res;
res.x = a.x ^ 0x8000; res.x = a.x ^ 0x8000;
return res; return res;
#endif
} }
HOSTDEVICE inline float16& operator+=(float16& a, const float16& b) { HOSTDEVICE inline float16& operator+=(float16& a, const float16& b) {
...@@ -550,6 +555,7 @@ HOSTDEVICE inline bool operator==(const float16& a, const float16& b) { ...@@ -550,6 +555,7 @@ HOSTDEVICE inline bool operator==(const float16& a, const float16& b) {
return __heq(half(a), half(b)); return __heq(half(a), half(b));
#else #else
return float(a) == float(b); return float(a) == float(b);
#endif
} }
HOSTDEVICE inline bool operator!=(const float16& a, const float16& b) { HOSTDEVICE inline bool operator!=(const float16& a, const float16& b) {
...@@ -557,6 +563,7 @@ HOSTDEVICE inline bool operator!=(const float16& a, const float16& b) { ...@@ -557,6 +563,7 @@ HOSTDEVICE inline bool operator!=(const float16& a, const float16& b) {
return __hne(half(a), half(b)); return __hne(half(a), half(b));
#else #else
return float(a) != float(b); return float(a) != float(b);
#endif
} }
HOSTDEVICE inline bool operator<(const float16& a, const float16& b) { HOSTDEVICE inline bool operator<(const float16& a, const float16& b) {
...@@ -564,6 +571,7 @@ HOSTDEVICE inline bool operator<(const float16& a, const float16& b) { ...@@ -564,6 +571,7 @@ HOSTDEVICE inline bool operator<(const float16& a, const float16& b) {
return __hlt(half(a), half(b)); return __hlt(half(a), half(b));
#else #else
return float(a) < float(b); return float(a) < float(b);
#endif
} }
HOSTDEVICE inline bool operator<=(const float16& a, const float16& b) { HOSTDEVICE inline bool operator<=(const float16& a, const float16& b) {
...@@ -571,6 +579,7 @@ HOSTDEVICE inline bool operator<=(const float16& a, const float16& b) { ...@@ -571,6 +579,7 @@ HOSTDEVICE inline bool operator<=(const float16& a, const float16& b) {
return __hle(half(a), half(b)); return __hle(half(a), half(b));
#else #else
return float(a) <= float(b); return float(a) <= float(b);
#endif
} }
HOSTDEVICE inline bool operator>(const float16& a, const float16& b) { HOSTDEVICE inline bool operator>(const float16& a, const float16& b) {
...@@ -578,6 +587,7 @@ HOSTDEVICE inline bool operator>(const float16& a, const float16& b) { ...@@ -578,6 +587,7 @@ HOSTDEVICE inline bool operator>(const float16& a, const float16& b) {
return __hgt(half(a), half(b)); return __hgt(half(a), half(b));
#else #else
return float(a) > float(b); return float(a) > float(b);
#endif
} }
HOSTDEVICE inline bool operator>=(const float16& a, const float16& b) { HOSTDEVICE inline bool operator>=(const float16& a, const float16& b) {
...@@ -585,6 +595,7 @@ HOSTDEVICE inline bool operator>=(const float16& a, const float16& b) { ...@@ -585,6 +595,7 @@ HOSTDEVICE inline bool operator>=(const float16& a, const float16& b) {
return __hge(half(a), half(b)); return __hge(half(a), half(b));
#else #else
return float(a) >= float(b); return float(a) >= float(b);
#endif
} }
// Arithmetic operators for float16 on ARMv8.2-A CPU // Arithmetic operators for float16 on ARMv8.2-A CPU
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或