]> git.ipfire.org Git - thirdparty/vectorscan.git/commitdiff
provide an {l,r}shift128_var() to fix immediate value build failure in loadu_maskz
authorKonstantinos Margaritis <konstantinos@vectorcamp.gr>
Wed, 21 Jul 2021 10:20:40 +0000 (10:20 +0000)
committerKonstantinos Margaritis <konstantinos@vectorcamp.gr>
Tue, 12 Oct 2021 08:51:34 +0000 (11:51 +0300)
src/util/supervector/arch/arm/impl.cpp
src/util/supervector/supervector.hpp

index 2c4cf3d67708e1badb5b7402a07661d7f7e7c5b0..e40b6a38c6a50f3eacf8473ec966e1c48b7615b4 100644 (file)
 template<>
 really_inline SuperVector<16>::SuperVector(SuperVector const &other)
 {
-       u.v128[0] = other.u.v128[0];
+  u.v128[0] = other.u.v128[0];
 }
 
 template<>
 really_inline SuperVector<16>::SuperVector(typename base_type::type const v)
 {
-       u.v128[0] = v;
+  u.v128[0] = v;
 };
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<int8x16_t>(int8x16_t const other)
 {
-       u.v128[0] = static_cast<int32x4_t>(other);
+  u.v128[0] = static_cast<int32x4_t>(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<uint8x16_t>(uint8x16_t const other)
 {
-       u.v128[0] = static_cast<int32x4_t>(other);
+  u.v128[0] = static_cast<int32x4_t>(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<int8_t>(int8_t const other)
 {
-       u.v128[0] = vdupq_n_s8(other);
+  u.v128[0] = vdupq_n_s8(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<uint8_t>(uint8_t const other)
 {
-       u.v128[0] = vdupq_n_u8(other);
+  u.v128[0] = vdupq_n_u8(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<int16_t>(int16_t const other)
 {
-       u.v128[0] = vdupq_n_s16(other);
+  u.v128[0] = vdupq_n_s16(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<uint16_t>(uint16_t const other)
 {
-       u.v128[0] = vdupq_n_u16(other);
+  u.v128[0] = vdupq_n_u16(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<int32_t>(int32_t const other)
 {
-       u.v128[0] = vdupq_n_s32(other);
+  u.v128[0] = vdupq_n_s32(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<uint32_t>(uint32_t const other)
 {
-       u.v128[0] = vdupq_n_u32(other);
+  u.v128[0] = vdupq_n_u32(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<int64_t>(int64_t const other)
 {
-       u.v128[0] = vdupq_n_s64(other);
+  u.v128[0] = vdupq_n_s64(other);
 }
 
 template<>
 template<>
 really_inline SuperVector<16>::SuperVector<uint64_t>(uint64_t const other)
 {
-       u.v128[0] = vdupq_n_u64(other);
+  u.v128[0] = vdupq_n_u64(other);
 }
 
 // Constants
@@ -195,22 +195,11 @@ really_inline typename SuperVector<16>::movemask_type SuperVector<16>::movemask(
 template <>
 really_inline typename SuperVector<16>::movemask_type SuperVector<16>::eqmask(SuperVector<16> const b) const
 {
-       return eq(b).movemask();
+  return eq(b).movemask();
 }
 
-#ifdef HS_OPTIMIZE
 template <>
-really_inline SuperVector<16> SuperVector<16>::operator>>(uint8_t const N) const
-{
-    if (N >= 16) {
-        return Zeroes();
-    } else {
-        return {vextq_s8((int16x8_t)u.v128[0], vdupq_n_u8(0), N)};
-    }
-}
-#else
-template <>
-really_inline SuperVector<16> SuperVector<16>::operator>>(uint8_t const N) const
+really_inline SuperVector<16> SuperVector<16>::rshift128_var(uint8_t const N) const
 {
     switch(N) {
     case 1: return {vextq_s8((int16x8_t)u.v128[0], vdupq_n_u8(0), 1)}; break;
@@ -233,21 +222,23 @@ really_inline SuperVector<16> SuperVector<16>::operator>>(uint8_t const N) const
     }
     return *this;
 }
-#endif
 
 #ifdef HS_OPTIMIZE
 template <>
-really_inline SuperVector<16> SuperVector<16>::operator<<(uint8_t const N) const
+really_inline SuperVector<16> SuperVector<16>::operator>>(uint8_t const N) const
 {
-    if (N == 0) {
-        return *this;
-    } else {
-        return {vextq_s8(vdupq_n_u8(0), (int16x8_t)u.v128[0], 16 - N)};
-    }
+    return {vextq_s8((int16x8_t)u.v128[0], vdupq_n_u8(0), N)};
 }
 #else
 template <>
-really_inline SuperVector<16> SuperVector<16>::operator<<(uint8_t const N) const
+really_inline SuperVector<16> SuperVector<16>::operator>>(uint8_t const N) const
+{
+    return rshift128_var(N);
+}
+#endif
+
+template <>
+really_inline SuperVector<16> SuperVector<16>::lshift128_var(uint8_t const N) const
 {
     switch(N) {
     case 1: return {vextq_s8(vdupq_n_u8(0), (int16x8_t)u.v128[0], 15)}; break;
@@ -270,6 +261,19 @@ really_inline SuperVector<16> SuperVector<16>::operator<<(uint8_t const N) const
     }
     return *this;
 }
+
+#ifdef HS_OPTIMIZE
+template <>
+really_inline SuperVector<16> SuperVector<16>::operator<<(uint8_t const N) const
+{
+    return {vextq_s8(vdupq_n_u8(0), (int16x8_t)u.v128[0], 16 - N)};
+}
+#else
+template <>
+really_inline SuperVector<16> SuperVector<16>::operator<<(uint8_t const N) const
+{
+    return lshift128_var(N);
+}
 #endif
 
 template <>
@@ -289,7 +293,7 @@ really_inline SuperVector<16> SuperVector<16>::load(void const *ptr)
 template <>
 really_inline SuperVector<16> SuperVector<16>::loadu_maskz(void const *ptr, uint8_t const len)
 {
-    SuperVector<16> mask = Ones() >> (16 -len);
+    SuperVector<16> mask = Ones().rshift128_var(16 -len);
     mask.print8("mask");
     SuperVector<16> v = loadu(ptr);
     v.print8("v");
@@ -310,27 +314,27 @@ really_inline SuperVector<16> SuperVector<16>::alignr(SuperVector<16> &other, in
 template<>
 really_inline SuperVector<16> SuperVector<16>::alignr(SuperVector<16> &other, int8_t offset)
 {
-       switch(offset) {
-       case 0: return other; break;
-       case 1: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 1)}; break;
-       case 2: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 2)}; break;
-       case 3: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 3)}; break;
-       case 4: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 4)}; break;
-       case 5: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 5)}; break;
-       case 6: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 6)}; break;
-       case 7: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 7)}; break;
-       case 8: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 8)}; break;
-       case 9: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 9)}; break;
-       case 10: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 10)}; break;
-       case 11: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 11)}; break;
-       case 12: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 12)}; break;
-       case 13: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 13)}; break;
-       case 14: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 14)}; break;
-       case 15: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 15)}; break;
-       case 16: return *this; break;
-       default: break;
-       }
-       return *this;
+  switch(offset) {
+  case 0: return other; break;
+  case 1: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 1)}; break;
+  case 2: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 2)}; break;
+  case 3: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 3)}; break;
+  case 4: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 4)}; break;
+  case 5: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 5)}; break;
+  case 6: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 6)}; break;
+  case 7: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 7)}; break;
+  case 8: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 8)}; break;
+  case 9: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 9)}; break;
+  case 10: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 10)}; break;
+  case 11: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 11)}; break;
+  case 12: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 12)}; break;
+  case 13: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 13)}; break;
+  case 14: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 14)}; break;
+  case 15: return {vextq_s8((int16x8_t) other.u.v128[0], (int16x8_t) u.v128[0], 15)}; break;
+  case 16: return *this; break;
+  default: break;
+  }
+  return *this;
 }
 #endif
 
@@ -348,32 +352,32 @@ really_inline SuperVector<16> SuperVector<16>::pshufb(SuperVector<16> b)
 template<>
 really_inline SuperVector<16> SuperVector<16>::lshift64(uint8_t const N)
 {
-       return {(m128)vshlq_n_s64(u.v128[0], N)};
+  return {(m128)vshlq_n_s64(u.v128[0], N)};
 }
 #else
 template<>
 really_inline SuperVector<16> SuperVector<16>::lshift64(uint8_t const N)
 {
-       switch(N) {
-       case 0: return {(m128)vshlq_n_s64(u.v128[0], 0)}; break;
-       case 1: return {(m128)vshlq_n_s64(u.v128[0], 1)}; break;
-       case 2: return {(m128)vshlq_n_s64(u.v128[0], 2)}; break;
-       case 3: return {(m128)vshlq_n_s64(u.v128[0], 3)}; break;
-       case 4: return {(m128)vshlq_n_s64(u.v128[0], 4)}; break;
-       case 5: return {(m128)vshlq_n_s64(u.v128[0], 5)}; break;
-       case 6: return {(m128)vshlq_n_s64(u.v128[0], 6)}; break;
-       case 7: return {(m128)vshlq_n_s64(u.v128[0], 7)}; break;
-       case 8: return {(m128)vshlq_n_s64(u.v128[0], 8)}; break;
-       case 9: return {(m128)vshlq_n_s64(u.v128[0], 9)}; break;
-       case 10: return {(m128)vshlq_n_s64(u.v128[0], 10)}; break;
-       case 11: return {(m128)vshlq_n_s64(u.v128[0], 11)}; break;
-       case 12: return {(m128)vshlq_n_s64(u.v128[0], 12)}; break;
-       case 13: return {(m128)vshlq_n_s64(u.v128[0], 13)}; break;
-       case 14: return {(m128)vshlq_n_s64(u.v128[0], 14)}; break;
-       case 15: return {(m128)vshlq_n_s64(u.v128[0], 15)}; break;
-       default: break;
-       }
-       return *this;
+  switch(N) {
+  case 0: return {(m128)vshlq_n_s64(u.v128[0], 0)}; break;
+  case 1: return {(m128)vshlq_n_s64(u.v128[0], 1)}; break;
+  case 2: return {(m128)vshlq_n_s64(u.v128[0], 2)}; break;
+  case 3: return {(m128)vshlq_n_s64(u.v128[0], 3)}; break;
+  case 4: return {(m128)vshlq_n_s64(u.v128[0], 4)}; break;
+  case 5: return {(m128)vshlq_n_s64(u.v128[0], 5)}; break;
+  case 6: return {(m128)vshlq_n_s64(u.v128[0], 6)}; break;
+  case 7: return {(m128)vshlq_n_s64(u.v128[0], 7)}; break;
+  case 8: return {(m128)vshlq_n_s64(u.v128[0], 8)}; break;
+  case 9: return {(m128)vshlq_n_s64(u.v128[0], 9)}; break;
+  case 10: return {(m128)vshlq_n_s64(u.v128[0], 10)}; break;
+  case 11: return {(m128)vshlq_n_s64(u.v128[0], 11)}; break;
+  case 12: return {(m128)vshlq_n_s64(u.v128[0], 12)}; break;
+  case 13: return {(m128)vshlq_n_s64(u.v128[0], 13)}; break;
+  case 14: return {(m128)vshlq_n_s64(u.v128[0], 14)}; break;
+  case 15: return {(m128)vshlq_n_s64(u.v128[0], 15)}; break;
+  default: break;
+  }
+  return *this;
 }
 #endif
 
@@ -381,45 +385,45 @@ really_inline SuperVector<16> SuperVector<16>::lshift64(uint8_t const N)
 template<>
 really_inline SuperVector<16> SuperVector<16>::rshift64(uint8_t const N)
 {
-       return {(m128)vshrq_n_s64(u.v128[0], N)};
+  return {(m128)vshrq_n_s64(u.v128[0], N)};
 }
 #else
 template<>
 really_inline SuperVector<16> SuperVector<16>::rshift64(uint8_t const N)
 {
-       switch(N) {
-       case 0: return {(m128)vshrq_n_s64(u.v128[0], 0)}; break;
-       case 1: return {(m128)vshrq_n_s64(u.v128[0], 1)}; break;
-       case 2: return {(m128)vshrq_n_s64(u.v128[0], 2)}; break;
-       case 3: return {(m128)vshrq_n_s64(u.v128[0], 3)}; break;
-       case 4: return {(m128)vshrq_n_s64(u.v128[0], 4)}; break;
-       case 5: return {(m128)vshrq_n_s64(u.v128[0], 5)}; break;
-       case 6: return {(m128)vshrq_n_s64(u.v128[0], 6)}; break;
-       case 7: return {(m128)vshrq_n_s64(u.v128[0], 7)}; break;
-       case 8: return {(m128)vshrq_n_s64(u.v128[0], 8)}; break;
-       case 9: return {(m128)vshrq_n_s64(u.v128[0], 9)}; break;
-       case 10: return {(m128)vshrq_n_s64(u.v128[0], 10)}; break;
-       case 11: return {(m128)vshrq_n_s64(u.v128[0], 11)}; break;
-       case 12: return {(m128)vshrq_n_s64(u.v128[0], 12)}; break;
-       case 13: return {(m128)vshrq_n_s64(u.v128[0], 13)}; break;
-       case 14: return {(m128)vshrq_n_s64(u.v128[0], 14)}; break;
-       case 15: return {(m128)vshrq_n_s64(u.v128[0], 15)}; break;
-       default: break;
-       }
-       return *this;
+  switch(N) {
+  case 0: return {(m128)vshrq_n_s64(u.v128[0], 0)}; break;
+  case 1: return {(m128)vshrq_n_s64(u.v128[0], 1)}; break;
+  case 2: return {(m128)vshrq_n_s64(u.v128[0], 2)}; break;
+  case 3: return {(m128)vshrq_n_s64(u.v128[0], 3)}; break;
+  case 4: return {(m128)vshrq_n_s64(u.v128[0], 4)}; break;
+  case 5: return {(m128)vshrq_n_s64(u.v128[0], 5)}; break;
+  case 6: return {(m128)vshrq_n_s64(u.v128[0], 6)}; break;
+  case 7: return {(m128)vshrq_n_s64(u.v128[0], 7)}; break;
+  case 8: return {(m128)vshrq_n_s64(u.v128[0], 8)}; break;
+  case 9: return {(m128)vshrq_n_s64(u.v128[0], 9)}; break;
+  case 10: return {(m128)vshrq_n_s64(u.v128[0], 10)}; break;
+  case 11: return {(m128)vshrq_n_s64(u.v128[0], 11)}; break;
+  case 12: return {(m128)vshrq_n_s64(u.v128[0], 12)}; break;
+  case 13: return {(m128)vshrq_n_s64(u.v128[0], 13)}; break;
+  case 14: return {(m128)vshrq_n_s64(u.v128[0], 14)}; break;
+  case 15: return {(m128)vshrq_n_s64(u.v128[0], 15)}; break;
+  default: break;
+  }
+  return *this;
 }
 #endif
 
 template<>
 really_inline SuperVector<16> SuperVector<16>::lshift128(uint8_t const N)
 {
-       return *this << N;
+  return *this << N;
 }
 
 template<>
 really_inline SuperVector<16> SuperVector<16>::rshift128(uint8_t const N)
 {
-       return *this >> N;
+  return *this >> N;
 }
 
 
index 0017592fe51fe0a0070daf75dc81f1f75c87d586..d245e0ca9b75413c4f30d6a035bf4d62bc3a886d 100644 (file)
@@ -200,6 +200,8 @@ public:
   SuperVector rshift64(uint8_t const N);
   SuperVector lshift128(uint8_t const N);
   SuperVector rshift128(uint8_t const N);
+  SuperVector lshift128_var(uint8_t const N) const;
+  SuperVector rshift128_var(uint8_t const N) const;
 
   // Constants
   static SuperVector Ones();