Fix batch_bool

Author: AntoinePrv

include/xsimd/arch/xsimd_avx.hpp

@@ -1019,13 +1019,13 @@ namespace xsimd
         namespace detail_avx
         {
             template <class A>
-            XSIMD_INLINE void maskstore(float* mem, batch_bool<float, A> const& mask, batch<float, A> const& src) noexcept
+            XSIMD_INLINE void maskstore(float* mem, batch<int32_t, A> const& mask, batch<float, A> const& src) noexcept
             {
                 _mm256_maskstore_ps(mem, mask, src);
             }
 
             template <class A>
-            XSIMD_INLINE void maskstore(double* mem, batch_bool<double, A> const& mask, batch<double, A> const& src) noexcept
+            XSIMD_INLINE void maskstore(double* mem, batch<int64_t, A> const& mask, batch<double, A> const& src) noexcept
             {
                 _mm256_maskstore_pd(mem, mask, src);
             }
@@ -1054,9 +1054,10 @@ namespace xsimd
             else
             {
                 using fp_t = sized_fp_t<sizeof(T)>;
+                using int_t = sized_int_t<sizeof(T)>;
                 detail_avx::maskstore<A>(
                     reinterpret_cast<fp_t*>(mem),
-                    mask.as_batch_bool(),
+                    bitwise_cast<int_t>(mask.as_batch()),
                     bitwise_cast<fp_t>(src));
             }
         }