WIP

Author: serge-sans-paille

include/xsimd/arch/xsimd_avx.hpp

@@ -1218,10 +1218,15 @@ namespace xsimd
         XSIMD_INLINE void
         store_complex_masked(std::complex<T>* mem, batch<std::complex<T>, A> const& src, batch_bool<T, A> mask, Mode mode, requires_arch<avx>) noexcept
         {
+            using mask_register_type = typename batch_bool<T, A>::register_type;
+            mask_register_type nmask = mask.to_native();
+            batch_bool<T, A> lo_mask = zip_lo(batch<T, A>(nmask), batch<T, A>(nmask)).to_native();
+            batch_bool<T, A> hi_mask = zip_hi(batch<T, A>(nmask), batch<T, A>(nmask)).to_native();
+
             auto src_lo = detail::complex_low(src, A{});
             auto src_hi = detail::complex_high(src, A{});
-            store_masked(reinterpret_cast<T*>(mem), src_lo, mask, mode, A{});
-            store_masked(reinterpret_cast<T*>(mem) + src.size, src_hi, mask, mode, A{});
+            store_masked(reinterpret_cast<T*>(mem), src_lo, lo_mask, mode, A{});
+            store_masked(reinterpret_cast<T*>(mem) + src.size, src_hi, hi_mask, mode, A{});
         }
 
         namespace detail