https://gcc.gnu.org/g:916f3b3802bf84083142149165da95c113d5d2e2

commit r13-8708-g916f3b3802bf84083142149165da95c113d5d2e2
Author: Matthias Kretz <m.kr...@gsi.de>
Date:   Wed Mar 27 13:41:25 2024 +0100

    libstdc++: Add masked ++/-- implementation for sizeof < 16
    
    This resolves further failures (-Wreturn-type warnings) and test
    failures for where-* tests targeting AVX-512.
    
    Signed-off-by: Matthias Kretz <m.kr...@gsi.de>
    
    libstdc++-v3/ChangeLog:
    
            * include/experimental/bits/simd_x86.h (_S_masked_unary):
            Cast inputs < 16 bytes to 16 byte vectors before calling the
            right subtraction builtin. Before returning, truncate to the
            return vector type.
    
    (cherry picked from commit a6c630c314b099f64d79055964d88b257459cf13)

Diff:
---
 libstdc++-v3/include/experimental/bits/simd_x86.h | 24 +++++++++++++----------
 1 file changed, 14 insertions(+), 10 deletions(-)

diff --git a/libstdc++-v3/include/experimental/bits/simd_x86.h 
b/libstdc++-v3/include/experimental/bits/simd_x86.h
index 81f46adb8ca2..79c246455ee4 100644
--- a/libstdc++-v3/include/experimental/bits/simd_x86.h
+++ b/libstdc++-v3/include/experimental/bits/simd_x86.h
@@ -3508,6 +3508,9 @@ template <typename _Abi, typename>
 #ifdef __clang__
            return __movm<_Np, _Tp>(__k._M_data) ? __v._M_data - __pm_one : 
__v._M_data;
 #else // __clang__
+           using _TV = __vector_type_t<_Tp, _Np>;
+           constexpr size_t __bytes = sizeof(__v) < 16 ? 16 : sizeof(__v);
+           constexpr size_t __width = __bytes / sizeof(_Tp);
            if constexpr (is_integral_v<_Tp>)
              {
                constexpr bool __lp64 = sizeof(long) == sizeof(long long);
@@ -3517,11 +3520,11 @@ template <typename _Abi, typename>
                              std::conditional_t<__lp64, long long, int>,
                              std::conditional_t<
                                std::is_same_v<_Ip, signed char>, char, _Ip>>;
-               const auto __value = __vector_bitcast<_Up>(__v._M_data);
+               const auto __value = __intrin_bitcast<__vector_type_t<_Up, 
__width>>(__v._M_data);
 #define _GLIBCXX_SIMD_MASK_SUB(_Sizeof, _Width, _Instr)                        
\
-  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__v) == _Width)               
\
-    return __vector_bitcast<_Tp>(__builtin_ia32_##_Instr##_mask(__value,       
\
-            __vector_broadcast<_Np>(_Up(__pm_one)), __value, __k._M_data))
+  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__value) == _Width)           
\
+    return __intrin_bitcast<_TV>(__builtin_ia32_##_Instr##_mask(__value,       
\
+            __vector_broadcast<__width>(_Up(__pm_one)), __value, __k._M_data))
                _GLIBCXX_SIMD_MASK_SUB(1, 64, psubb512);
                _GLIBCXX_SIMD_MASK_SUB(1, 32, psubb256);
                _GLIBCXX_SIMD_MASK_SUB(1, 16, psubb128);
@@ -3538,16 +3541,17 @@ template <typename _Abi, typename>
              }
            else
              {
+               const auto __value = __intrin_bitcast<__vector_type_t<_Tp, 
__width>>(__v._M_data);
 #define _GLIBCXX_SIMD_MASK_SUB_512(_Sizeof, _Width, _Instr)                    
\
-  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__v) == _Width)               
\
+  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__value) == _Width)           
\
     return __builtin_ia32_##_Instr##_mask(                                     
\
-            __v._M_data, __vector_broadcast<_Np>(_Tp(__pm_one)), __v._M_data, \
+            __value, __vector_broadcast<__width>(_Tp(__pm_one)), __value,     \
             __k._M_data, _MM_FROUND_CUR_DIRECTION)
 #define _GLIBCXX_SIMD_MASK_SUB(_Sizeof, _Width, _Instr)                        
\
-  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__v) == _Width)               
\
-    return __builtin_ia32_##_Instr##_mask(                                     
\
-            __v._M_data, __vector_broadcast<_Np>(_Tp(__pm_one)), __v._M_data, \
-            __k._M_data)
+  if constexpr (sizeof(_Tp) == _Sizeof && sizeof(__value) == _Width)           
\
+    return __intrin_bitcast<_TV>(__builtin_ia32_##_Instr##_mask(               
\
+            __value, __vector_broadcast<__width>(_Tp(__pm_one)), __value,     \
+            __k._M_data))
                _GLIBCXX_SIMD_MASK_SUB_512(4, 64, subps512);
                _GLIBCXX_SIMD_MASK_SUB(4, 32, subps256);
                _GLIBCXX_SIMD_MASK_SUB(4, 16, subps128);

Reply via email to