Question

我对

的行为有点好奇

_mm512_mask_i32scatter_epi32(void* base_addr, __mask16 mask, __512i idx, __512i data, int scale)

此内在函数应使用来自idx寄存器的32位索引从数据寄存器中分散32位整数。只有在屏蔽寄存器中设置了相应的位时，才会存储一个值。在官方文档之后，值从base_addr开始存储，并带有与idx寄存器相对应的偏移量。比例用于缩放偏移量。

我的数据寄存器（data_reg）如下所示：

[ 0] = 4    [ 4] = 0    [ 8] = 0    [12] = 0    
[ 1] = 5    [ 5] = 0    [ 9] = 0    [13] = 0    
[ 2] = 4    [ 6] = 0    [10] = 0    [14] = 0    
[ 3] = 0    [ 7] = 0    [11] = 0    [15] = 0

索引寄存器（idx_reg）如下所示：

[ 0] = 0    [ 4] = 8    [ 8] = 16   [12] = 24   
[ 1] = 2    [ 5] = 10   [ 9] = 18   [13] = 26   
[ 2] = 4    [ 6] = 12   [10] = 20   [14] = 28   
[ 3] = 6    [ 7] = 14   [11] = 22   [15] = 30

掩码寄存器（mask_reg）如下所示：

[ 0] = 1    [ 4] = 0    [ 8] = 0    [12] = 0    
[ 1] = 1    [ 5] = 0    [ 9] = 0    [13] = 0    
[ 2] = 1    [ 6] = 0    [10] = 0    [14] = 0    
[ 3] = 0    [ 7] = 0    [11] = 0    [15] = 0

我称之为内在：

_mm512_mask_i32scatter_epi32( result_array, mask_reg, idx_reg, data_reg, 1);

结果数据（result_array）如下所示：

[ 0] = 327684   [ 4] = 0    [ 8] = 0    [12] = 0    
[ 1] = 4        [ 5] = 0    [ 9] = 0    [13] = 0    
[ 2] = 0        [ 6] = 0    [10] = 0    [14] = 0    
[ 3] = 0        [ 7] = 0    [11] = 0    [15] = 0

但它应该是这样的：

[ 0] = 4        [ 4] = 4    [ 8] = 0    [12] = 0    
[ 1] = 0        [ 5] = 0    [ 9] = 0    [13] = 0    
[ 2] = 5        [ 6] = 0    [10] = 0    [14] = 0    
[ 3] = 0        [ 7] = 0    [11] = 0    [15] = 0

我错过了什么或这种行为有点奇怪吗？

此致

Answer 1

索引是 byte 偏移，而不是元素偏移，即你需要将idx_reg中的索引乘以sizeof(int32_t)，或者通过sizeof(int32_t)作为scale参数而不是1。

另见this related question。

使用avx-512的分散存储

1 个答案: