feat: Add batch::raw

Author: AntoinePrv

include/xsimd/types/xsimd_batch.hpp

@@ -193,6 +193,8 @@ namespace xsimd
 
         XSIMD_INLINE T first() const noexcept;
 
+        XSIMD_INLINE register_type raw() const noexcept;
+
         // comparison operators. Defined as friend to enable automatic
         // conversion of parameters from scalar to batch, at the cost of using a
         // proxy implementation from details::.
@@ -355,6 +357,8 @@ namespace xsimd
 
         XSIMD_INLINE bool first() const noexcept;
 
+        XSIMD_INLINE register_type raw() const noexcept;
+
         // mask operations
         XSIMD_INLINE uint64_t mask() const noexcept;
         XSIMD_INLINE static batch_bool from_mask(uint64_t mask) noexcept;
@@ -836,6 +840,15 @@ namespace xsimd
         return kernel::first(*this, A {});
     }
 
+    /**
+     * Cast to the underlying native intrinsic register type.
+     */
+    template <class T, class A>
+    XSIMD_INLINE auto batch<T, A>::raw() const noexcept -> register_type
+    {
+        return static_cast<register_type>(*this);
+    }
+
     /******************************
      * batch comparison operators *
      ******************************/
@@ -1167,6 +1180,15 @@ namespace xsimd
         return kernel::first(*this, A {});
     }
 
+    /**
+     * Cast to the underlying native intrinsic register type.
+     */
+    template <class T, class A>
+    XSIMD_INLINE auto batch_bool<T, A>::raw() const noexcept -> register_type
+    {
+        return static_cast<register_type>(*this);
+    }
+
     /***********************************
      * batch_bool comparison operators *
      ***********************************/

test/test_batch.cpp

@@ -100,6 +100,16 @@ struct batch_test
         CHECK_EQ(res_dump.str(), b_dump.str());
     }
 
+    void test_register_conversion() const {
+        batch_type b1 = batch_type::load_unaligned(lhs.data());
+
+        auto b_native = b1.raw();
+        CHECK(std::is_same<decltype(b_native), typename batch_type::register_type>::value);
+
+        batch_type b2 = b_native;  // Converting ctor
+        CHECK_EQ(b1, b2);
+    }
+
     void test_load_store() const
     {
         array_type res;