AI/rocm-systems
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Guard the OCML rounded operations instead

Browse Source 
Instead of commenting all these functions out, guard the functions with a macro OCML_BASIC_ROUNDED_OPERATIONS.
This commit is contained in:
Aaron Enye Shi
2018-11-06 16:32:14 +00:00
parent 7223277aa3
commit e59c33250a
3 changed files with 236 additions and 216 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/hip/hcc_detail/math_functions.h
+178 -168
View File
@@ -514,69 +514,73 @@ float __exp10f(float x) { return __ocml_exp10_f32(x); }
__DEVICE__
inline
float __expf(float x) { return __ocml_exp_f32(x); }
// __DEVICE__
// inline
// float __fadd_rd(float x, float y) { return __ocml_add_rtn_f32(x, y); }
// __DEVICE__
// inline
// float __fadd_rn(float x, float y) { return __ocml_add_rte_f32(x, y); }
// __DEVICE__
// inline
// float __fadd_ru(float x, float y) { return __ocml_add_rtp_f32(x, y); }
// __DEVICE__
// inline
// float __fadd_rz(float x, float y) { return __ocml_add_rtz_f32(x, y); }
// __DEVICE__
// inline
// float __fdiv_rd(float x, float y) { return __ocml_div_rtn_f32(x, y); }
// __DEVICE__
// inline
// float __fdiv_rn(float x, float y) { return __ocml_div_rte_f32(x, y); }
// __DEVICE__
// inline
// float __fdiv_ru(float x, float y) { return __ocml_div_rtp_f32(x, y); }
// __DEVICE__
// inline
// float __fdiv_rz(float x, float y) { return __ocml_div_rtz_f32(x, y); }
#if defined OCML_BASIC_ROUNDED_OPERATIONS
__DEVICE__
inline
float __fadd_rd(float x, float y) { return __ocml_add_rtn_f32(x, y); }
__DEVICE__
inline
float __fadd_rn(float x, float y) { return __ocml_add_rte_f32(x, y); }
__DEVICE__
inline
float __fadd_ru(float x, float y) { return __ocml_add_rtp_f32(x, y); }
__DEVICE__
inline
float __fadd_rz(float x, float y) { return __ocml_add_rtz_f32(x, y); }
__DEVICE__
inline
float __fdiv_rd(float x, float y) { return __ocml_div_rtn_f32(x, y); }
__DEVICE__
inline
float __fdiv_rn(float x, float y) { return __ocml_div_rte_f32(x, y); }
__DEVICE__
inline
float __fdiv_ru(float x, float y) { return __ocml_div_rtp_f32(x, y); }
__DEVICE__
inline
float __fdiv_rz(float x, float y) { return __ocml_div_rtz_f32(x, y); }
#endif
__DEVICE__
inline
float __fdividef(float x, float y) { return x / y; }
// __DEVICE__
// inline
// float __fmaf_rd(float x, float y, float z)
// {
// return __ocml_fma_rtn_f32(x, y, z);
// }
// __DEVICE__
// inline
// float __fmaf_rn(float x, float y, float z)
// {
// return __ocml_fma_rte_f32(x, y, z);
// }
// __DEVICE__
// inline
// float __fmaf_ru(float x, float y, float z)
// {
// return __ocml_fma_rtp_f32(x, y, z);
// }
// __DEVICE__
// inline
// float __fmaf_rz(float x, float y, float z)
// {
// return __ocml_fma_rtz_f32(x, y, z);
// }
// __DEVICE__
// inline
// float __fmul_rd(float x, float y) { return __ocml_mul_rtn_f32(x, y); }
// __DEVICE__
// inline
// float __fmul_rn(float x, float y) { return __ocml_mul_rte_f32(x, y); }
// __DEVICE__
// inline
// float __fmul_ru(float x, float y) { return __ocml_mul_rtp_f32(x, y); }
// __DEVICE__
// inline
// float __fmul_rz(float x, float y) { return __ocml_mul_rtz_f32(x, y); }
#if defined OCML_BASIC_ROUNDED_OPERATIONS
__DEVICE__
inline
float __fmaf_rd(float x, float y, float z)
{
return __ocml_fma_rtn_f32(x, y, z);
}
__DEVICE__
inline
float __fmaf_rn(float x, float y, float z)
{
return __ocml_fma_rte_f32(x, y, z);
}
__DEVICE__
inline
float __fmaf_ru(float x, float y, float z)
{
return __ocml_fma_rtp_f32(x, y, z);
}
__DEVICE__
inline
float __fmaf_rz(float x, float y, float z)
{
return __ocml_fma_rtz_f32(x, y, z);
}
__DEVICE__
inline
float __fmul_rd(float x, float y) { return __ocml_mul_rtn_f32(x, y); }
__DEVICE__
inline
float __fmul_rn(float x, float y) { return __ocml_mul_rte_f32(x, y); }
__DEVICE__
inline
float __fmul_ru(float x, float y) { return __ocml_mul_rtp_f32(x, y); }
__DEVICE__
inline
float __fmul_rz(float x, float y) { return __ocml_mul_rtz_f32(x, y); }
#endif
__DEVICE__
inline
float __frcp_rd(float x) { return __llvm_amdgcn_rcp_f32(x); }
@@ -592,30 +596,32 @@ float __frcp_rz(float x) { return __llvm_amdgcn_rcp_f32(x); }
__DEVICE__
inline
float __frsqrt_rn(float x) { return __llvm_amdgcn_rsq_f32(x); }
// __DEVICE__
// inline
// float __fsqrt_rd(float x) { return __ocml_sqrt_rtn_f32(x); }
// __DEVICE__
// inline
// float __fsqrt_rn(float x) { return __ocml_sqrt_rte_f32(x); }
// __DEVICE__
// inline
// float __fsqrt_ru(float x) { return __ocml_sqrt_rtp_f32(x); }
// __DEVICE__
// inline
// float __fsqrt_rz(float x) { return __ocml_sqrt_rtz_f32(x); }
// __DEVICE__
// inline
// float __fsub_rd(float x, float y) { return __ocml_sub_rtn_f32(x, y); }
// __DEVICE__
// inline
// float __fsub_rn(float x, float y) { return __ocml_sub_rte_f32(x, y); }
// __DEVICE__
// inline
// float __fsub_ru(float x, float y) { return __ocml_sub_rtp_f32(x, y); }
// __DEVICE__
// inline
// float __fsub_rz(float x, float y) { return __ocml_sub_rtz_f32(x, y); }
#if defined OCML_BASIC_ROUNDED_OPERATIONS
__DEVICE__
inline
float __fsqrt_rd(float x) { return __ocml_sqrt_rtn_f32(x); }
__DEVICE__
inline
float __fsqrt_rn(float x) { return __ocml_sqrt_rte_f32(x); }
__DEVICE__
inline
float __fsqrt_ru(float x) { return __ocml_sqrt_rtp_f32(x); }
__DEVICE__
inline
float __fsqrt_rz(float x) { return __ocml_sqrt_rtz_f32(x); }
__DEVICE__
inline
float __fsub_rd(float x, float y) { return __ocml_sub_rtn_f32(x, y); }
__DEVICE__
inline
float __fsub_rn(float x, float y) { return __ocml_sub_rte_f32(x, y); }
__DEVICE__
inline
float __fsub_ru(float x, float y) { return __ocml_sub_rtp_f32(x, y); }
__DEVICE__
inline
float __fsub_rz(float x, float y) { return __ocml_sub_rtz_f32(x, y); }
#endif
__DEVICE__
inline
float __log10f(float x) { return __ocml_log10_f32(x); }
@@ -1034,42 +1040,44 @@ double yn(int n, double x)
}
// BEGIN INTRINSICS
// __DEVICE__
// inline
// double __dadd_rd(double x, double y) { return __ocml_add_rtn_f64(x, y); }
// __DEVICE__
// inline
// double __dadd_rn(double x, double y) { return __ocml_add_rte_f64(x, y); }
// __DEVICE__
// inline
// double __dadd_ru(double x, double y) { return __ocml_add_rtp_f64(x, y); }
// __DEVICE__
// inline
// double __dadd_rz(double x, double y) { return __ocml_add_rtz_f64(x, y); }
// __DEVICE__
// inline
// double __ddiv_rd(double x, double y) { return __ocml_div_rtn_f64(x, y); }
// __DEVICE__
// inline
// double __ddiv_rn(double x, double y) { return __ocml_div_rte_f64(x, y); }
// __DEVICE__
// inline
// double __ddiv_ru(double x, double y) { return __ocml_div_rtp_f64(x, y); }
// __DEVICE__
// inline
// double __ddiv_rz(double x, double y) { return __ocml_div_rtz_f64(x, y); }
// __DEVICE__
// inline
// double __dmul_rd(double x, double y) { return __ocml_mul_rtn_f64(x, y); }
// __DEVICE__
// inline
// double __dmul_rn(double x, double y) { return __ocml_mul_rte_f64(x, y); }
// __DEVICE__
// inline
// double __dmul_ru(double x, double y) { return __ocml_mul_rtp_f64(x, y); }
// __DEVICE__
// inline
// double __dmul_rz(double x, double y) { return __ocml_mul_rtz_f64(x, y); }
#if defined OCML_BASIC_ROUNDED_OPERATIONS
__DEVICE__
inline
double __dadd_rd(double x, double y) { return __ocml_add_rtn_f64(x, y); }
__DEVICE__
inline
double __dadd_rn(double x, double y) { return __ocml_add_rte_f64(x, y); }
__DEVICE__
inline
double __dadd_ru(double x, double y) { return __ocml_add_rtp_f64(x, y); }
__DEVICE__
inline
double __dadd_rz(double x, double y) { return __ocml_add_rtz_f64(x, y); }
__DEVICE__
inline
double __ddiv_rd(double x, double y) { return __ocml_div_rtn_f64(x, y); }
__DEVICE__
inline
double __ddiv_rn(double x, double y) { return __ocml_div_rte_f64(x, y); }
__DEVICE__
inline
double __ddiv_ru(double x, double y) { return __ocml_div_rtp_f64(x, y); }
__DEVICE__
inline
double __ddiv_rz(double x, double y) { return __ocml_div_rtz_f64(x, y); }
__DEVICE__
inline
double __dmul_rd(double x, double y) { return __ocml_mul_rtn_f64(x, y); }
__DEVICE__
inline
double __dmul_rn(double x, double y) { return __ocml_mul_rte_f64(x, y); }
__DEVICE__
inline
double __dmul_ru(double x, double y) { return __ocml_mul_rtp_f64(x, y); }
__DEVICE__
inline
double __dmul_rz(double x, double y) { return __ocml_mul_rtz_f64(x, y); }
#endif
__DEVICE__
inline
double __drcp_rd(double x) { return __llvm_amdgcn_rcp_f64(x); }
@@ -1082,54 +1090,56 @@ double __drcp_ru(double x) { return __llvm_amdgcn_rcp_f64(x); }
__DEVICE__
inline
double __drcp_rz(double x) { return __llvm_amdgcn_rcp_f64(x); }
// __DEVICE__
// inline
// double __dsqrt_rd(double x) { return __ocml_sqrt_rtn_f64(x); }
// __DEVICE__
// inline
// double __dsqrt_rn(double x) { return __ocml_sqrt_rte_f64(x); }
// __DEVICE__
// inline
// double __dsqrt_ru(double x) { return __ocml_sqrt_rtp_f64(x); }
// __DEVICE__
// inline
// double __dsqrt_rz(double x) { return __ocml_sqrt_rtz_f64(x); }
// __DEVICE__
// inline
// double __dsub_rd(double x, double y) { return __ocml_sub_rtn_f64(x, y); }
// __DEVICE__
// inline
// double __dsub_rn(double x, double y) { return __ocml_sub_rte_f64(x, y); }
// __DEVICE__
// inline
// double __dsub_ru(double x, double y) { return __ocml_sub_rtp_f64(x, y); }
// __DEVICE__
// inline
// double __dsub_rz(double x, double y) { return __ocml_sub_rtz_f64(x, y); }
// __DEVICE__
// inline
// double __fma_rd(double x, double y, double z)
// {
// return __ocml_fma_rtn_f64(x, y, z);
// }
// __DEVICE__
// inline
// double __fma_rn(double x, double y, double z)
// {
// return __ocml_fma_rte_f64(x, y, z);
// }
// __DEVICE__
// inline
// double __fma_ru(double x, double y, double z)
// {
// return __ocml_fma_rtp_f64(x, y, z);
// }
// __DEVICE__
// inline
// double __fma_rz(double x, double y, double z)
// {
// return __ocml_fma_rtz_f64(x, y, z);
// }
#if defined OCML_BASIC_ROUNDED_OPERATIONS
__DEVICE__
inline
double __dsqrt_rd(double x) { return __ocml_sqrt_rtn_f64(x); }
__DEVICE__
inline
double __dsqrt_rn(double x) { return __ocml_sqrt_rte_f64(x); }
__DEVICE__
inline
double __dsqrt_ru(double x) { return __ocml_sqrt_rtp_f64(x); }
__DEVICE__
inline
double __dsqrt_rz(double x) { return __ocml_sqrt_rtz_f64(x); }
__DEVICE__
inline
double __dsub_rd(double x, double y) { return __ocml_sub_rtn_f64(x, y); }
__DEVICE__
inline
double __dsub_rn(double x, double y) { return __ocml_sub_rte_f64(x, y); }
__DEVICE__
inline
double __dsub_ru(double x, double y) { return __ocml_sub_rtp_f64(x, y); }
__DEVICE__
inline
double __dsub_rz(double x, double y) { return __ocml_sub_rtz_f64(x, y); }
__DEVICE__
inline
double __fma_rd(double x, double y, double z)
{
return __ocml_fma_rtn_f64(x, y, z);
}
__DEVICE__
inline
double __fma_rn(double x, double y, double z)
{
return __ocml_fma_rte_f64(x, y, z);
}
__DEVICE__
inline
double __fma_ru(double x, double y, double z)
{
return __ocml_fma_rtp_f64(x, y, z);
}
__DEVICE__
inline
double __fma_rz(double x, double y, double z)
{
return __ocml_fma_rtz_f64(x, y, z);
}
#endif
// END INTRINSICS
// END DOUBLE