如何从__m512i类型中提取指定位置的8位整数（AVX-512）

AVX-512指令集没有直接对应_mm_extract_epi8/_mm256_extract_epi8的内置函数，可根据索引是编译期常量还是运行期变量选择不同实现：

1. 索引为编译期常量 #

这种场景下编译器能生成最优指令，推荐通过提取对应64位块+移位的方式实现：

#include <immintrin.h>

inline int extract_m512i_epi8(__m512i a, const int idx) {
    const int qword_idx = idx / 8;    // 计算目标字节所在的64位块索引
    const int byte_off = idx % 8;     // 块内字节偏移
    uint64_t qword = _mm512_extract_epi64(a, qword_idx);
    // 提取目标字节并做符号扩展为int类型
    return (int)((int8_t)((qword >> (byte_off * 8)) & 0xFF));
}

编译时编译器会直接优化掉索引计算，生成无额外开销的提取+移位指令。

2. 索引为运行期变量 #

推荐使用_mm512_permutexvar_epi8将目标字节重排到向量首位置，再提取低字节，这是寄存器内操作，避免内存访问开销：

#include <immintrin.h>

inline int extract_m512i_epi8_var(__m512i a, int idx) {
    __m512i idx_vec = _mm512_set1_epi8((char)idx);
    // 将目标字节移到向量第一个位置
    __m512i permuted = _mm512_permutexvar_epi8(idx_vec, a);
    // 提取首字节并转换为int
    return (int)_mm_cvtsi128_si32(_mm512_castsi512_si128(permuted));
}

也可以将向量存储到内存后直接读取对应字节，但这种方式会引入内存IO开销，仅在需要批量提取多个字节时考虑。

内容的提问来源于stack exchange，提问作者KaraUL