0

对于像这个例子这样的情况

for (int i = 16; i < n; i++) 
    a[i] += a[i-16];

如何确定向量长度以确保可以对这个循环进行向量化？以下是正确的方法吗？

// Determine target cpu architecture's vector register bit-size
// E.g., Intel AVX-512 has 512-bit vector registers
int register_size = 512

// Modern machines are 8 bits
int byte_size = 8

// Determine array type size
int my_array[n];
auto array_type_size = sizeof(int) * byte_size   // 4 bytes * 8 bits = 32 bits

// Divide register size by array type size
auto vector_length = register_size / array_type_size    // = 16

这意味着在英特尔 AVX-512 上对 int 类型的数组进行向量化将导致寄存器中的向量长度为​​ 16，从而使上述示例可以安全地进行向量化。

这种方法正确吗？如果是这样，有没有办法在这个架构上使用更短的向量？例如，强制一个长度为 4 的向量，以便可以对下面的示例进行向量化

for (int i = 4; i < n; i++) 
    a[i] += a[i-4];
4

1 回答 1

0

英特尔提供了一系列 SIMD 功能。您可以使用从 64 位 mmx 寄存器到 512 zmm 寄存器。您也可以很好地使用这些寄存器的下半部分：您可以使用 256 位 ymm，甚至可以使用别名的下半部分并使用 128 位 xmm。无论您的循环范围或大小如何，您的编译器都会在发布模式下尽可能多地对循环进行矢量化。如果您想自己执行此操作，您可以编辑编译器标志或编写自己的程序集，以实现小幅但显着的速度提升。要“强制”编译器以某种方式向量化，将更改编译器标志以禁止使用 AVX-512，或者可能使用编译器将在其文档中包含的一些预处理器指令。不使用带有矢量化的整个寄存器没有固有的问题。

于 2020-03-02T07:29:40.007 回答