编写AVX优化函数时，能否将uint64_t转为const __m256i_u？

• 绝对不能直接强制转换：uint64_t是64位无符号整数，而__m256i_u是256位SIMD寄存器类型，两者内存宽度、语义完全不同，直接强制转换会导致内存越界或数据错乱，属于未定义行为。

• 正确做法是使用AVX intrinsics进行数据打包/加载：

  • 批量加载uint64_t到__m256i_u：若你有包含4个uint64_t的数组，可通过_mm256_loadu_si256((const __m256i_u*)arr)（非对齐内存）或_mm256_load_si256（对齐内存）完成加载。
  • 单个uint64_t广播到__m256i_u：用_mm256_set1_epi64x((int64_t)val)将单个64位值填充到整个256位寄存器，无符号场景下数值正确时无需担心类型转换问题。
  • 从__m256i_u提取uint64_t：用_mm256_extract_epi64(reg, idx)获取寄存器中第idx位的64位值，或用_mm256_storeu_si256((__m256i_u*)arr, reg)将寄存器内容存入uint64_t数组。

• 额外提示：

  • 永远不要手动通过指针强制转换来在标量类型和SIMD寄存器类型间转数据，必须用官方提供的intrinsic函数，否则会破坏编译器优化逻辑，引发不可预料的问题。
  • 若函数仅接收单个uint64_t参数，AVX的SIMD并行优势无法发挥，建议设计成处理uint64_t数组，一次性处理多个元素才能体现AVX的性能提升。

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