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,17 @@ 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_BATCH_EQ(b1, b2);
+    }
+
     void test_load_store() const
     {
         array_type res;
@@ -1161,6 +1172,11 @@ TEST_CASE_TEMPLATE("[batch]", B, BATCH_TYPES)
         Test.test_stream_dump();
     }
 
+    SUBCASE("register_conversion")
+    {
+        Test.test_register_conversion();
+    }
+
     SUBCASE("load_store")
     {
         Test.test_load_store();