如何将__m128i的16个8位元素转换为两个__m256i的32位元素?
无需AVX-512:将__m128i的16个8位元素转换为两个__m256i寄存器
完全可以做到,用AVX2指令集就能实现,全程不需要AVX-512,避免降频问题。核心思路是把16个8位元素拆分为两组各8个,分别将每组8位元素扩展为32位,存入两个__m256i寄存器。
具体实现步骤(以无符号8位元素为例)
假设你已经将16字节数据加载到__m128i寄存器中:
#include <immintrin.h> // 输入:包含16个无符号8位元素的128位寄存器 __m128i vec_8bit = _mm_loadu_si128((const __m128i*)your_data_ptr); // 1. 处理低8个8位元素:直接扩展为8个32位元素,存入第一个__m256i __m256i vec32_low = _mm256_cvtepu8_epi32(vec_8bit); // 2. 处理高8个8位元素:先将原寄存器右移8字节,把高8位元素移到低字节位置 __m128i vec_8bit_high_part = _mm_srli_si128(vec_8bit, 8); // 再扩展为8个32位元素,存入第二个__m256i __m256i vec32_high = _mm256_cvtepu8_epi32(vec_8bit_high_part);
关键指令说明
_mm256_cvtepu8_epi32:AVX2指令,接收一个__m128i寄存器,提取其低8个无符号8位元素,每个元素零扩展为32位,最终输出一个包含8个32位元素的__m256i寄存器。_mm_srli_si128:SSE指令,将__m128i寄存器向右移动指定字节数(这里是8字节),把原寄存器的高8个8位元素移到低字节区域,方便后续扩展。
如果处理的是有符号8位元素,只需将_mm256_cvtepu8_epi32替换为_mm256_cvtepi8_epi32,该指令会进行符号扩展而非零扩展。
这种方案效率极高,每个扩展操作都是单指令完成,完全避开AVX-512指令集,不会触发CPU降频。
内容的提问来源于stack exchange,提问作者user997112
相关产品推荐
相关产品推荐

