Fix interaction between make_sized_batch_t and batches of complex

Fix #1049

Author: serge-sans-paille

include/xsimd/types/xsimd_batch.hpp

@@ -1442,7 +1442,7 @@ namespace xsimd
             using type = void;
         };
 
-        template <typename T, class Arch, bool BatchExists = xsimd::types::has_simd_register<T, Arch>::value>
+        template <typename T, class Arch, bool BatchExists = xsimd::has_simd_register<T, Arch>::value>
         struct batch_trait;
 
         template <typename T, class Arch>

include/xsimd/types/xsimd_traits.hpp

@@ -35,6 +35,11 @@ namespace xsimd
     {
     };
 
+    template <class T, class A>
+    struct has_simd_register<std::complex<T>, A> : has_simd_register<T, A>
+    {
+    };
+
     namespace detail
     {
         template <class T, bool>

test/test_arch.cpp

@@ -142,26 +142,34 @@ TEST_CASE("[multi arch support]")
     {
         using batch4f = xsimd::make_sized_batch_t<float, 4>;
         using batch2d = xsimd::make_sized_batch_t<double, 2>;
+        using batch4c = xsimd::make_sized_batch_t<std::complex<float>, 4>;
+        using batch2z = xsimd::make_sized_batch_t<std::complex<double>, 2>;
         using batch4i32 = xsimd::make_sized_batch_t<int32_t, 4>;
         using batch4u32 = xsimd::make_sized_batch_t<uint32_t, 4>;
 
         using batch8f = xsimd::make_sized_batch_t<float, 8>;
         using batch4d = xsimd::make_sized_batch_t<double, 4>;
+        using batch8c = xsimd::make_sized_batch_t<std::complex<float>, 8>;
+        using batch4z = xsimd::make_sized_batch_t<std::complex<double>, 4>;
         using batch8i32 = xsimd::make_sized_batch_t<int32_t, 8>;
         using batch8u32 = xsimd::make_sized_batch_t<uint32_t, 8>;
 
 #if XSIMD_WITH_SSE2 || XSIMD_WITH_NEON || XSIMD_WITH_NEON64 || XSIMD_WITH_SVE || (XSIMD_WITH_RVV && XSIMD_RVV_BITS == 128)
         CHECK_EQ(4, size_t(batch4f::size));
+        CHECK_EQ(4, size_t(batch4c::size));
         CHECK_EQ(4, size_t(batch4i32::size));
         CHECK_EQ(4, size_t(batch4u32::size));
 
         CHECK_UNARY(bool(std::is_same<float, batch4f::value_type>::value));
+        CHECK_UNARY(bool(std::is_same<std::complex<float>, batch4c::value_type>::value));
         CHECK_UNARY(bool(std::is_same<int32_t, batch4i32::value_type>::value));
         CHECK_UNARY(bool(std::is_same<uint32_t, batch4u32::value_type>::value));
 
 #if XSIMD_WITH_SSE2 || XSIMD_WITH_NEON64 || XSIMD_WITH_SVE || XSIMD_WITH_RVV
         CHECK_EQ(2, size_t(batch2d::size));
+        CHECK_EQ(2, size_t(batch2z::size));
         CHECK_UNARY(bool(std::is_same<double, batch2d::value_type>::value));
+        CHECK_UNARY(bool(std::is_same<std::complex<double>, batch2z::value_type>::value));
 #else
         CHECK_UNARY(bool(std::is_same<void, batch2d>::value));
 #endif
@@ -170,6 +178,8 @@ TEST_CASE("[multi arch support]")
 #if !XSIMD_WITH_AVX && !XSIMD_WITH_FMA3 && !(XSIMD_WITH_SVE && XSIMD_SVE_BITS == 256) && !(XSIMD_WITH_RVV && XSIMD_RVV_BITS == 256)
         CHECK_UNARY(bool(std::is_same<void, batch8f>::value));
         CHECK_UNARY(bool(std::is_same<void, batch4d>::value));
+        CHECK_UNARY(bool(std::is_same<void, batch8c>::value));
+        CHECK_UNARY(bool(std::is_same<void, batch4z>::value));
         CHECK_UNARY(bool(std::is_same<void, batch8i32>::value));
         CHECK_UNARY(bool(std::is_same<void, batch8u32>::value));
 #else
@@ -178,10 +188,15 @@ TEST_CASE("[multi arch support]")
         CHECK_EQ(8, size_t(batch8u32::size));
         CHECK_EQ(4, size_t(batch4d::size));
 
+        CHECK_EQ(8, size_t(batch8c::size));
+        CHECK_EQ(4, size_t(batch4z::size));
+
         CHECK_UNARY(bool(std::is_same<float, batch8f::value_type>::value));
         CHECK_UNARY(bool(std::is_same<double, batch4d::value_type>::value));
         CHECK_UNARY(bool(std::is_same<int32_t, batch8i32::value_type>::value));
         CHECK_UNARY(bool(std::is_same<uint32_t, batch8u32::value_type>::value));
+        CHECK_UNARY(bool(std::is_same<std::complex<float>, batch8c::value_type>::value));
+        CHECK_UNARY(bool(std::is_same<std::complex<double>, batch4z::value_type>::value));
 #endif
     }