Add fma function with float and _Float16 arguments

This commit is contained in:
Yaxun Sam Liu
2018-08-10 11:12:53 -04:00
والد 2a5a2c66c4
کامیت bd622a4b4a
2فایلهای تغییر یافته به همراه152 افزوده شده و 0 حذف شده
@@ -1166,6 +1166,16 @@ long long llabs(long long x)
#endif
// END INTEGER
__DEVICE__
inline _Float16 fma(_Float16 x, _Float16 y, _Float16 z) {
return __ocml_fma_f16(x, y, z);
}
__DEVICE__
inline float fma(float x, float y, float z) {
return fmaf(x, y, z);
}
#pragma push_macro("__DEF_FLOAT_FUN")
#pragma push_macro("__DEF_FLOAT_FUN2")
#pragma push_macro("__DEF_FLOAT_FUN2I")