Add fma function with float and _Float16 arguments

[ROCm/hip commit: bd622a4b4a]
This commit is contained in:
Yaxun Sam Liu
2018-08-10 11:12:53 -04:00
parent bf9fa30a60
commit 2d4040f75d
2 changed files with 152 additions and 0 deletions
@@ -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")