如何在AVX512寄存器中高效生成uint64_t类型的0-7序列？

最高效的方式是直接在寄存器内生成序列，完全避免内存访问带来的缓存缺失风险，具体可以通过AVX512的向量初始化指令实现，无需从内存加载数据。

具体代码实现 #

#include <immintrin.h>

__m512i generate_uint64_sequence() {
    // _mm512_setr_epi64x 按从低到高的顺序初始化向量元素，正好对应目标序列[0,1,2,3,4,5,6,7]
    return _mm512_setr_epi64x(0, 1, 2, 3, 4, 5, 6, 7);
}

为什么这比内存加载更优 #

• 无内存访问：该指令由编译器直接编译为寄存器级操作，不需要读取内存中的预存序列，彻底规避缓存缺失的可能性。
• 更低延迟：寄存器内操作的延迟远低于内存加载，尤其是在缓存未命中的场景下，性能差距会被放大。
• 更高吞吐量：无需占用内存带宽，能更高效地利用CPU的向量执行单元。

补充：通用序列生成方案 #

如果需要生成起始值或步长不同的算术序列，可以结合广播和加法指令实现，比如生成[2,3,4,5,6,7,8,9]：

__m512i generate_custom_sequence() {
    // 广播起始值2到所有64位元素
    __m512i base = _mm512_set1_epi64x(2);
    // 生成递增的偏移向量[0,1,2,3,4,5,6,7]
    __m512i offsets = _mm512_setr_epi64x(0, 1, 2, 3, 4, 5, 6, 7);
    // 相加得到目标序列
    return _mm512_add_epi64(base, offsets);
}

但对于固定的[0,1,2,3,4,5,6,7]序列，直接使用_mm512_setr_epi64x初始化是最简洁且高效的方案。

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