[SYCL] Support stdlib function abs, labs, llabs in device library (#3364)

Signed-off-by: haonanya <[email protected]>

Author: haonanya

libdevice/cmath_wrapper.cpp

@@ -9,6 +9,16 @@
 #include "device_math.h"
 
 #ifdef __SPIR__
+
+DEVICE_EXTERN_C
+int abs(int x) { return __devicelib_abs(x); }
+
+DEVICE_EXTERN_C
+long int labs(long int x) { return __devicelib_labs(x); }
+
+DEVICE_EXTERN_C
+long long int llabs(long long int x) { return __devicelib_llabs(x); }
+
 DEVICE_EXTERN_C
 div_t div(int x, int y) { return __devicelib_div(x, y); }
 

libdevice/device_math.h

@@ -13,6 +13,15 @@
 #ifdef __SPIR__
 #include <cstdlib>
 
+DEVICE_EXTERN_C
+int __devicelib_abs(int x);
+
+DEVICE_EXTERN_C
+long int __devicelib_labs(long int x);
+
+DEVICE_EXTERN_C
+long long int __devicelib_llabs(long long int x);
+
 DEVICE_EXTERN_C
 div_t __devicelib_div(int x, int y);
 

libdevice/fallback-cmath.cpp

@@ -16,6 +16,15 @@
 // TODO: generate the DeviceLibFuncMap in sycl-post-link.cpp automatically
 // during the build based on libdevice to avoid manually sync.
 
+DEVICE_EXTERN_C
+int __devicelib_abs(int x) { return x < 0 ? -x : x; }
+
+DEVICE_EXTERN_C
+long int __devicelib_labs(long int x) { return x < 0 ? -x : x; }
+
+DEVICE_EXTERN_C
+long long int __devicelib_llabs(long long int x) { return x < 0 ? -x : x; }
+
 DEVICE_EXTERN_C
 div_t __devicelib_div(int x, int y) { return {x / y, x % y}; }
 

llvm/tools/sycl-post-link/SYCLDeviceLibReqMask.cpp

@@ -24,6 +24,7 @@ namespace {
 // Please update SDLMap if any item is added to or removed from
 // fallback device libraries in libdevice.
 SYCLDeviceLibFuncMap SDLMap = {
+    {"__devicelib_abs", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_acosf", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_acoshf", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_asinf", DeviceLibExt::cl_intel_devicelib_math},
@@ -46,9 +47,11 @@ SYCLDeviceLibFuncMap SDLMap = {
     {"__devicelib_frexpf", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_hypotf", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_ilogbf", DeviceLibExt::cl_intel_devicelib_math},
+    {"__devicelib_labs", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_ldiv", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_ldexpf", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_lgammaf", DeviceLibExt::cl_intel_devicelib_math},
+    {"__devicelib_llabs", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_lldiv", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_log10f", DeviceLibExt::cl_intel_devicelib_math},
     {"__devicelib_log1pf", DeviceLibExt::cl_intel_devicelib_math},

sycl/include/CL/sycl/builtins.hpp

@@ -1545,6 +1545,10 @@ detail::enable_if_t<detail::is_genfloatf<T>::value, T> tan(T x) __NOEXC {
 
 #ifdef __SYCL_DEVICE_ONLY__
 extern "C" {
+extern SYCL_EXTERNAL int abs(int x);
+extern SYCL_EXTERNAL long int labs(long int x);
+extern SYCL_EXTERNAL long long int llabs(long long int x);
+
 extern SYCL_EXTERNAL div_t div(int x, int y);
 extern SYCL_EXTERNAL ldiv_t ldiv(long int x, long int y);
 extern SYCL_EXTERNAL lldiv_t lldiv(long long int x, long long int y);