Add shifts

Author: AntoinePrv

include/xsimd/types/xsimd_batch_constant.hpp

@@ -194,6 +194,14 @@ namespace xsimd
         {
             constexpr T operator()(T x, T y) const { return x ^ y; }
         };
+        struct binary_rshift
+        {
+            constexpr T operator()(T x, T y) const { return x >> y; }
+        };
+        struct binary_lshift
+        {
+            constexpr T operator()(T x, T y) const { return x << y; }
+        };
 
         template <class F, class SelfPack, class OtherPack, std::size_t... Indices>
         static constexpr batch_constant<T, A, F()(std::tuple_element<Indices, SelfPack>::type::value, std::tuple_element<Indices, OtherPack>::type::value)...>
@@ -227,6 +235,8 @@ namespace xsimd
         MAKE_BINARY_OP(&, binary_and)
         MAKE_BINARY_OP(|, binary_or)
         MAKE_BINARY_OP(^, binary_xor)
+        MAKE_BINARY_OP(<<, binary_lshift)
+        MAKE_BINARY_OP(>>, binary_rshift)
 
 #undef MAKE_BINARY_OP
 

test/test_batch_constant.cpp

@@ -136,6 +136,14 @@ struct constant_batch_test
         constexpr auto n12_lxor_n3 = n12 ^ n3;
         static_assert(std::is_same<decltype(n12_lxor_n3), decltype(n15)>::value, "n12 ^ n3 == n15");
 
+        constexpr auto n96 = xsimd::make_batch_constant<value_type, constant<96>, arch_type>();
+        constexpr auto n12_lshift_n3 = n12 << n3;
+        static_assert(std::is_same<decltype(n12_lshift_n3), decltype(n96)>::value, "n12 << n3 == n96");
+
+        constexpr auto n1 = xsimd::make_batch_constant<value_type, constant<1>, arch_type>();
+        constexpr auto n12_rshift_n3 = n12 >> n3;
+        static_assert(std::is_same<decltype(n12_rshift_n3), decltype(n1)>::value, "n12 >> n3 == n1");
+
         constexpr auto n12_uadd = +n12;
         static_assert(std::is_same<decltype(n12_uadd), decltype(n12)>::value, "+n12 == n12");