457 lines
14 KiB
ReStructuredText
457 lines
14 KiB
ReStructuredText
==========================
|
|
Auto-Vectorization in LLVM
|
|
==========================
|
|
|
|
.. contents::
|
|
:local:
|
|
|
|
LLVM has two vectorizers: The :ref:`Loop Vectorizer <loop-vectorizer>`,
|
|
which operates on Loops, and the :ref:`SLP Vectorizer
|
|
<slp-vectorizer>`. These vectorizers
|
|
focus on different optimization opportunities and use different techniques.
|
|
The SLP vectorizer merges multiple scalars that are found in the code into
|
|
vectors while the Loop Vectorizer widens instructions in loops
|
|
to operate on multiple consecutive iterations.
|
|
|
|
Both the Loop Vectorizer and the SLP Vectorizer are enabled by default.
|
|
|
|
.. _loop-vectorizer:
|
|
|
|
The Loop Vectorizer
|
|
===================
|
|
|
|
Usage
|
|
-----
|
|
|
|
The Loop Vectorizer is enabled by default, but it can be disabled
|
|
through clang using the command line flag:
|
|
|
|
.. code-block:: console
|
|
|
|
$ clang ... -fno-vectorize file.c
|
|
|
|
Command line flags
|
|
^^^^^^^^^^^^^^^^^^
|
|
|
|
The loop vectorizer uses a cost model to decide on the optimal vectorization factor
|
|
and unroll factor. However, users of the vectorizer can force the vectorizer to use
|
|
specific values. Both 'clang' and 'opt' support the flags below.
|
|
|
|
Users can control the vectorization SIMD width using the command line flag "-force-vector-width".
|
|
|
|
.. code-block:: console
|
|
|
|
$ clang -mllvm -force-vector-width=8 ...
|
|
$ opt -loop-vectorize -force-vector-width=8 ...
|
|
|
|
Users can control the unroll factor using the command line flag "-force-vector-interleave"
|
|
|
|
.. code-block:: console
|
|
|
|
$ clang -mllvm -force-vector-interleave=2 ...
|
|
$ opt -loop-vectorize -force-vector-interleave=2 ...
|
|
|
|
Pragma loop hint directives
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
The ``#pragma clang loop`` directive allows loop vectorization hints to be
|
|
specified for the subsequent for, while, do-while, or c++11 range-based for
|
|
loop. The directive allows vectorization and interleaving to be enabled or
|
|
disabled. Vector width as well as interleave count can also be manually
|
|
specified. The following example explicitly enables vectorization and
|
|
interleaving:
|
|
|
|
.. code-block:: c++
|
|
|
|
#pragma clang loop vectorize(enable) interleave(enable)
|
|
while(...) {
|
|
...
|
|
}
|
|
|
|
The following example implicitly enables vectorization and interleaving by
|
|
specifying a vector width and interleaving count:
|
|
|
|
.. code-block:: c++
|
|
|
|
#pragma clang loop vectorize_width(2) interleave_count(2)
|
|
for(...) {
|
|
...
|
|
}
|
|
|
|
See the Clang
|
|
`language extensions
|
|
<https://clang.llvm.org/docs/LanguageExtensions.html#extensions-for-loop-hint-optimizations>`_
|
|
for details.
|
|
|
|
Diagnostics
|
|
-----------
|
|
|
|
Many loops cannot be vectorized including loops with complicated control flow,
|
|
unvectorizable types, and unvectorizable calls. The loop vectorizer generates
|
|
optimization remarks which can be queried using command line options to identify
|
|
and diagnose loops that are skipped by the loop-vectorizer.
|
|
|
|
Optimization remarks are enabled using:
|
|
|
|
``-Rpass=loop-vectorize`` identifies loops that were successfully vectorized.
|
|
|
|
``-Rpass-missed=loop-vectorize`` identifies loops that failed vectorization and
|
|
indicates if vectorization was specified.
|
|
|
|
``-Rpass-analysis=loop-vectorize`` identifies the statements that caused
|
|
vectorization to fail. If in addition ``-fsave-optimization-record`` is
|
|
provided, multiple causes of vectorization failure may be listed (this behavior
|
|
might change in the future).
|
|
|
|
Consider the following loop:
|
|
|
|
.. code-block:: c++
|
|
|
|
#pragma clang loop vectorize(enable)
|
|
for (int i = 0; i < Length; i++) {
|
|
switch(A[i]) {
|
|
case 0: A[i] = i*2; break;
|
|
case 1: A[i] = i; break;
|
|
default: A[i] = 0;
|
|
}
|
|
}
|
|
|
|
The command line ``-Rpass-missed=loop-vectorize`` prints the remark:
|
|
|
|
.. code-block:: console
|
|
|
|
no_switch.cpp:4:5: remark: loop not vectorized: vectorization is explicitly enabled [-Rpass-missed=loop-vectorize]
|
|
|
|
And the command line ``-Rpass-analysis=loop-vectorize`` indicates that the
|
|
switch statement cannot be vectorized.
|
|
|
|
.. code-block:: console
|
|
|
|
no_switch.cpp:4:5: remark: loop not vectorized: loop contains a switch statement [-Rpass-analysis=loop-vectorize]
|
|
switch(A[i]) {
|
|
^
|
|
|
|
To ensure line and column numbers are produced include the command line options
|
|
``-gline-tables-only`` and ``-gcolumn-info``. See the Clang `user manual
|
|
<https://clang.llvm.org/docs/UsersManual.html#options-to-emit-optimization-reports>`_
|
|
for details
|
|
|
|
Features
|
|
--------
|
|
|
|
The LLVM Loop Vectorizer has a number of features that allow it to vectorize
|
|
complex loops.
|
|
|
|
Loops with unknown trip count
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer supports loops with an unknown trip count.
|
|
In the loop below, the iteration ``start`` and ``finish`` points are unknown,
|
|
and the Loop Vectorizer has a mechanism to vectorize loops that do not start
|
|
at zero. In this example, 'n' may not be a multiple of the vector width, and
|
|
the vectorizer has to execute the last few iterations as scalar code. Keeping
|
|
a scalar copy of the loop increases the code size.
|
|
|
|
.. code-block:: c++
|
|
|
|
void bar(float *A, float* B, float K, int start, int end) {
|
|
for (int i = start; i < end; ++i)
|
|
A[i] *= B[i] + K;
|
|
}
|
|
|
|
Runtime Checks of Pointers
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
In the example below, if the pointers A and B point to consecutive addresses,
|
|
then it is illegal to vectorize the code because some elements of A will be
|
|
written before they are read from array B.
|
|
|
|
Some programmers use the 'restrict' keyword to notify the compiler that the
|
|
pointers are disjointed, but in our example, the Loop Vectorizer has no way of
|
|
knowing that the pointers A and B are unique. The Loop Vectorizer handles this
|
|
loop by placing code that checks, at runtime, if the arrays A and B point to
|
|
disjointed memory locations. If arrays A and B overlap, then the scalar version
|
|
of the loop is executed.
|
|
|
|
.. code-block:: c++
|
|
|
|
void bar(float *A, float* B, float K, int n) {
|
|
for (int i = 0; i < n; ++i)
|
|
A[i] *= B[i] + K;
|
|
}
|
|
|
|
|
|
Reductions
|
|
^^^^^^^^^^
|
|
|
|
In this example the ``sum`` variable is used by consecutive iterations of
|
|
the loop. Normally, this would prevent vectorization, but the vectorizer can
|
|
detect that 'sum' is a reduction variable. The variable 'sum' becomes a vector
|
|
of integers, and at the end of the loop the elements of the array are added
|
|
together to create the correct result. We support a number of different
|
|
reduction operations, such as addition, multiplication, XOR, AND and OR.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int *A, int n) {
|
|
unsigned sum = 0;
|
|
for (int i = 0; i < n; ++i)
|
|
sum += A[i] + 5;
|
|
return sum;
|
|
}
|
|
|
|
We support floating point reduction operations when `-ffast-math` is used.
|
|
|
|
Inductions
|
|
^^^^^^^^^^
|
|
|
|
In this example the value of the induction variable ``i`` is saved into an
|
|
array. The Loop Vectorizer knows to vectorize induction variables.
|
|
|
|
.. code-block:: c++
|
|
|
|
void bar(float *A, int n) {
|
|
for (int i = 0; i < n; ++i)
|
|
A[i] = i;
|
|
}
|
|
|
|
If Conversion
|
|
^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer is able to "flatten" the IF statement in the code and
|
|
generate a single stream of instructions. The Loop Vectorizer supports any
|
|
control flow in the innermost loop. The innermost loop may contain complex
|
|
nesting of IFs, ELSEs and even GOTOs.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int *A, int *B, int n) {
|
|
unsigned sum = 0;
|
|
for (int i = 0; i < n; ++i)
|
|
if (A[i] > B[i])
|
|
sum += A[i] + 5;
|
|
return sum;
|
|
}
|
|
|
|
Pointer Induction Variables
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
This example uses the "accumulate" function of the standard c++ library. This
|
|
loop uses C++ iterators, which are pointers, and not integer indices.
|
|
The Loop Vectorizer detects pointer induction variables and can vectorize
|
|
this loop. This feature is important because many C++ programs use iterators.
|
|
|
|
.. code-block:: c++
|
|
|
|
int baz(int *A, int n) {
|
|
return std::accumulate(A, A + n, 0);
|
|
}
|
|
|
|
Reverse Iterators
|
|
^^^^^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer can vectorize loops that count backwards.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int *A, int n) {
|
|
for (int i = n; i > 0; --i)
|
|
A[i] +=1;
|
|
}
|
|
|
|
Scatter / Gather
|
|
^^^^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer can vectorize code that becomes a sequence of scalar instructions
|
|
that scatter/gathers memory.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int * A, int * B, int n) {
|
|
for (intptr_t i = 0; i < n; ++i)
|
|
A[i] += B[i * 4];
|
|
}
|
|
|
|
In many situations the cost model will inform LLVM that this is not beneficial
|
|
and LLVM will only vectorize such code if forced with "-mllvm -force-vector-width=#".
|
|
|
|
Vectorization of Mixed Types
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer can vectorize programs with mixed types. The Vectorizer
|
|
cost model can estimate the cost of the type conversion and decide if
|
|
vectorization is profitable.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int *A, char *B, int n) {
|
|
for (int i = 0; i < n; ++i)
|
|
A[i] += 4 * B[i];
|
|
}
|
|
|
|
Global Structures Alias Analysis
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
Access to global structures can also be vectorized, with alias analysis being
|
|
used to make sure accesses don't alias. Run-time checks can also be added on
|
|
pointer access to structure members.
|
|
|
|
Many variations are supported, but some that rely on undefined behaviour being
|
|
ignored (as other compilers do) are still being left un-vectorized.
|
|
|
|
.. code-block:: c++
|
|
|
|
struct { int A[100], K, B[100]; } Foo;
|
|
|
|
int foo() {
|
|
for (int i = 0; i < 100; ++i)
|
|
Foo.A[i] = Foo.B[i] + 100;
|
|
}
|
|
|
|
Vectorization of function calls
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
The Loop Vectorizer can vectorize intrinsic math functions.
|
|
See the table below for a list of these functions.
|
|
|
|
+-----+-----+---------+
|
|
| pow | exp | exp2 |
|
|
+-----+-----+---------+
|
|
| sin | cos | sqrt |
|
|
+-----+-----+---------+
|
|
| log |log2 | log10 |
|
|
+-----+-----+---------+
|
|
|fabs |floor| ceil |
|
|
+-----+-----+---------+
|
|
|fma |trunc|nearbyint|
|
|
+-----+-----+---------+
|
|
| | | fmuladd |
|
|
+-----+-----+---------+
|
|
|
|
Note that the optimizer may not be able to vectorize math library functions
|
|
that correspond to these intrinsics if the library calls access external state
|
|
such as "errno". To allow better optimization of C/C++ math library functions,
|
|
use "-fno-math-errno".
|
|
|
|
The loop vectorizer knows about special instructions on the target and will
|
|
vectorize a loop containing a function call that maps to the instructions. For
|
|
example, the loop below will be vectorized on Intel x86 if the SSE4.1 roundps
|
|
instruction is available.
|
|
|
|
.. code-block:: c++
|
|
|
|
void foo(float *f) {
|
|
for (int i = 0; i != 1024; ++i)
|
|
f[i] = floorf(f[i]);
|
|
}
|
|
|
|
Partial unrolling during vectorization
|
|
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
Modern processors feature multiple execution units, and only programs that contain a
|
|
high degree of parallelism can fully utilize the entire width of the machine.
|
|
The Loop Vectorizer increases the instruction level parallelism (ILP) by
|
|
performing partial-unrolling of loops.
|
|
|
|
In the example below the entire array is accumulated into the variable 'sum'.
|
|
This is inefficient because only a single execution port can be used by the processor.
|
|
By unrolling the code the Loop Vectorizer allows two or more execution ports
|
|
to be used simultaneously.
|
|
|
|
.. code-block:: c++
|
|
|
|
int foo(int *A, int n) {
|
|
unsigned sum = 0;
|
|
for (int i = 0; i < n; ++i)
|
|
sum += A[i];
|
|
return sum;
|
|
}
|
|
|
|
The Loop Vectorizer uses a cost model to decide when it is profitable to unroll loops.
|
|
The decision to unroll the loop depends on the register pressure and the generated code size.
|
|
|
|
Epilogue Vectorization
|
|
^^^^^^^^^^^^^^^^^^^^^^
|
|
|
|
When vectorizing a loop, often a scalar remainder (epilogue) loop is necessary
|
|
to execute tail iterations of the loop if the loop trip count is unknown or it
|
|
does not evenly divide the vectorization and unroll factors. When the
|
|
vectorization and unroll factors are large, it's possible for loops with smaller
|
|
trip counts to end up spending most of their time in the scalar (rather than
|
|
the vector) code. In order to address this issue, the inner loop vectorizer is
|
|
enhanced with a feature that allows it to vectorize epilogue loops with a
|
|
vectorization and unroll factor combination that makes it more likely for small
|
|
trip count loops to still execute in vectorized code. The diagram below shows
|
|
the CFG for a typical epilogue vectorized loop with runtime checks. As
|
|
illustrated the control flow is structured in a way that avoids duplicating the
|
|
runtime pointer checks and optimizes the path length for loops that have very
|
|
small trip counts.
|
|
|
|
.. image:: epilogue-vectorization-cfg.png
|
|
|
|
Performance
|
|
-----------
|
|
|
|
This section shows the execution time of Clang on a simple benchmark:
|
|
`gcc-loops <https://github.com/llvm/llvm-test-suite/tree/master/SingleSource/UnitTests/Vectorizer>`_.
|
|
This benchmarks is a collection of loops from the GCC autovectorization
|
|
`page <http://gcc.gnu.org/projects/tree-ssa/vectorization.html>`_ by Dorit Nuzman.
|
|
|
|
The chart below compares GCC-4.7, ICC-13, and Clang-SVN with and without loop vectorization at -O3, tuned for "corei7-avx", running on a Sandybridge iMac.
|
|
The Y-axis shows the time in msec. Lower is better. The last column shows the geomean of all the kernels.
|
|
|
|
.. image:: gcc-loops.png
|
|
|
|
And Linpack-pc with the same configuration. Result is Mflops, higher is better.
|
|
|
|
.. image:: linpack-pc.png
|
|
|
|
Ongoing Development Directions
|
|
------------------------------
|
|
|
|
.. toctree::
|
|
:hidden:
|
|
|
|
Proposals/VectorizationPlan
|
|
|
|
:doc:`Proposals/VectorizationPlan`
|
|
Modeling the process and upgrading the infrastructure of LLVM's Loop Vectorizer.
|
|
|
|
.. _slp-vectorizer:
|
|
|
|
The SLP Vectorizer
|
|
==================
|
|
|
|
Details
|
|
-------
|
|
|
|
The goal of SLP vectorization (a.k.a. superword-level parallelism) is
|
|
to combine similar independent instructions
|
|
into vector instructions. Memory accesses, arithmetic operations, comparison
|
|
operations, PHI-nodes, can all be vectorized using this technique.
|
|
|
|
For example, the following function performs very similar operations on its
|
|
inputs (a1, b1) and (a2, b2). The basic-block vectorizer may combine these
|
|
into vector operations.
|
|
|
|
.. code-block:: c++
|
|
|
|
void foo(int a1, int a2, int b1, int b2, int *A) {
|
|
A[0] = a1*(a1 + b1);
|
|
A[1] = a2*(a2 + b2);
|
|
A[2] = a1*(a1 + b1);
|
|
A[3] = a2*(a2 + b2);
|
|
}
|
|
|
|
The SLP-vectorizer processes the code bottom-up, across basic blocks, in search of scalars to combine.
|
|
|
|
Usage
|
|
------
|
|
|
|
The SLP Vectorizer is enabled by default, but it can be disabled
|
|
through clang using the command line flag:
|
|
|
|
.. code-block:: console
|
|
|
|
$ clang -fno-slp-vectorize file.c
|