从C中的两个ymms填充一个zmm答案

【问题标题】：fill a zmm from two ymms in C从C中的两个ymms填充一个zmm
【发布时间】：2021-11-03 04:18:20
【问题描述】：

我想知道用简单包装 (zmm0 = {ymm1,ymm0}) 从两个 m256is 加载 m512i 的最佳方法是什么。我知道 ymm0 是 zmm0 的低位，但不确定我是否可以使用内在函数在 C 中利用它。在 C 中实现这一目标的最佳方法是什么？

【问题讨论】：

标签： c intrinsics avx2 avx512

【解决方案1】：

奇怪，在 Intel 的内在函数指南中似乎没有 256->512 版本的 _mm256_set_m128i。也许是因为每个 AVX512 内在函数都必须有一个 _mask_ 版本？不，还有_mm512_set_epi32，所以这很奇怪。

您可以将_mm512_cast 一个到__m512i 和vinserti32x8 另一个放入其中。（或 64x4，如果不屏蔽，则选择无关紧要。）

#include <immintrin.h>

__m256i merge256(__m128i lo, __m128i hi){
        //return _mm256_set_m128i(hi, lo);
        return _mm256_set_m128i(hi, lo);
}

#ifdef __AVX512F__
__m512i merge512(__m256i lo, __m256i hi){
    __m512i base = _mm512_castsi256_si512(lo);  // upper half is don't-care
    return _mm512_inserti32x8(base, hi, 1);     // insert hi as new upper half
    
//        return _mm512_set_m256i(b, a);  // doesn't exist in GCC, clang, ICC, or MSVC
}
#endif

Demo on Godbolt，还包括 128->256 和 _mm256_set_m128i(hi, lo)

对于这些示例，我将 arg 顺序定义为 lo, hi。您可能更愿意将其定义为 hi, lo 以匹配 _mm_set（而不是 setr）内在函数。

【讨论】：