aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorShow Liu <show.liu@linaro.org>2015-06-26 17:29:22 +0800
committerGil Pitney <gil.pitney@linaro.org>2015-06-26 16:27:01 +0000
commitdc931e9a05042727d3dd2e82f18cd580e6680bb5 (patch)
tree7ed5fcaedb74f1d593fefc4e252bd7a02e0cf866
parent1145df21bed854b764d974b2c9eae56364857f0a (diff)
downloadshamrock-dc931e9a05042727d3dd2e82f18cd580e6680bb5.tar.gz
Tixed the mad_sat() builtin function crash issue
-rw-r--r--src/builtins/mad_sat.cl73
1 files changed, 73 insertions, 0 deletions
diff --git a/src/builtins/mad_sat.cl b/src/builtins/mad_sat.cl
index ac79a86..e3f8e37 100644
--- a/src/builtins/mad_sat.cl
+++ b/src/builtins/mad_sat.cl
@@ -27,6 +27,79 @@
*****************************************************************************/
#include "cpu.h"
+_CLC_OVERLOAD _CLC_INLINE char mad_sat(char a, char b, char c)
+{
+ short tmp = (short)a * (short)b + (short)c;
+
+ if (tmp > (short)CHAR_MAX) return CHAR_MAX;
+ if (tmp < (short)CHAR_MIN) return CHAR_MIN;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE uchar mad_sat(uchar a, uchar b, uchar c)
+{
+ ushort tmp = (ushort)a * (ushort)b + (ushort)c;
+
+ if (tmp > (ushort)UCHAR_MAX) return UCHAR_MAX;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE short mad_sat(short a, short b, short c)
+{
+ int tmp = (int)a * (int)b + (int)c;
+
+ if (tmp > (int)SHRT_MAX) return SHRT_MAX;
+ if (tmp < (int)SHRT_MIN) return SHRT_MIN;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE ushort mad_sat(ushort a, ushort b, ushort c)
+{
+ uint tmp = (uint)a * (uint)b + (uint)c;
+
+ if (tmp > (uint)USHRT_MAX) return USHRT_MAX;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE int mad_sat(int a, int b, int c)
+{
+ long tmp = (long)a * (long)b + (long)c;
+
+ if (tmp > (long)INT_MAX) return INT_MAX;
+ if (tmp < (long)INT_MIN) return INT_MIN;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE uint mad_sat(uint a, uint b, uint c)
+{
+ ulong tmp = (ulong)a * (ulong)b + (ulong)c;
+
+ if (tmp > (ulong)UINT_MAX) return UINT_MAX;
+ return tmp;
+}
+
+_CLC_OVERLOAD _CLC_INLINE long mad_sat(long a, long b, long c)
+{
+ if (a > 0 && b > 0 && a > (LONG_MAX/b)) return LONG_MAX;
+ if (a > 0 && b < 0 && b < (LONG_MIN/a)) return LONG_MIN;
+ if (a < 0 && b > 0 && a < (LONG_MIN/b)) return LONG_MIN;
+ if (a < 0 && b < 0 && b < (LONG_MAX/a))
+ {
+ ulong tmp = a * b + c;
+ if(tmp == 0)
+ return tmp;
+ else
+ return LONG_MAX;
+ }
+ return add_sat(a*b, c);
+}
+
+_CLC_OVERLOAD _CLC_INLINE ulong mad_sat(ulong a, ulong b, ulong c)
+{
+ if (a > (ULONG_MAX/b)) return ULONG_MAX;
+ return add_sat(a*b, c);
+}
+
TERNARY_VEC_DEF(char, char, mad_sat, mad_sat)
TERNARY_VEC_DEF(uchar, uchar, mad_sat, mad_sat)
TERNARY_VEC_DEF(short, short, mad_sat, mad_sat)