2012-12-19 15:22:24 +08:00
==========================
Auto-Vectorization in LLVM
==========================
2012-12-21 06:42:20 +08:00
.. contents ::
:local:
LLVM has two vectorizers: The :ref: `Loop Vectorizer <loop-vectorizer>` ,
which operates on Loops, and the :ref:`Basic Block Vectorizer
<bb-vectorizer> `, which optimizes straight-line code. These vectorizers
focus on different optimization opportunities and use different techniques.
The BB vectorizer merges multiple scalars that are found in the code into
vectors while the Loop Vectorizer widens instructions in the original loop
to operate on multiple consecutive loop iterations.
.. _loop-vectorizer:
2012-12-19 15:22:24 +08:00
The Loop Vectorizer
===================
2012-12-20 02:04:44 +08:00
Usage
2012-12-20 10:40:45 +08:00
-----
2012-12-20 02:04:44 +08:00
2012-12-20 10:23:25 +08:00
LLVM's Loop Vectorizer is now available and will be useful for many people.
2012-12-19 15:22:24 +08:00
It is not enabled by default, but can be enabled through clang using the
command line flag:
.. code-block :: console
2012-12-20 02:02:36 +08:00
$ clang -fvectorize -O3 file.c
If the `` -fvectorize `` flag is used then the loop vectorizer will be enabled
when running with `` -O3 `` , `` -O2 `` . When `` -Os `` is used, the loop vectorizer
will only vectorize loops that do not require a major increase in code size.
2012-12-19 15:22:24 +08:00
We plan to enable the Loop Vectorizer by default as part of the LLVM 3.3 release.
2013-01-05 01:49:45 +08:00
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-unroll"
.. code-block :: console
$ clang -mllvm -force-vector-unroll=2 ...
$ opt -loop-vectorize -force-vector-unroll=2 ...
2012-12-19 15:22:24 +08:00
Features
2012-12-20 10:40:45 +08:00
--------
2012-12-19 15:22:24 +08:00
The LLVM Loop Vectorizer has a number of features that allow it to vectorize
complex loops.
Loops with unknown trip count
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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
2012-12-20 10:23:25 +08:00
at zero. In this example, 'n' may not be a multiple of the vector width, and
2012-12-19 15:22:24 +08:00
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) {
2012-12-21 06:47:41 +08:00
for (int i = start; i < end; ++i)
A[i] *= B[i] + K;
2012-12-19 15:22:24 +08:00
}
Runtime Checks of Pointers
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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
2012-12-21 06:59:36 +08:00
of the loop is executed.
2012-12-19 15:22:24 +08:00
.. code-block :: c++
void bar(float *A, float* B, float K, int n) {
2012-12-21 06:47:41 +08:00
for (int i = 0; i < n; ++i)
A[i] *= B[i] + K;
2012-12-19 15:22:24 +08:00
}
Reductions
2012-12-20 10:40:45 +08:00
^^^^^^^^^^
2012-12-19 15:22:24 +08:00
2012-12-21 06:59:36 +08:00
In this example the `` sum `` variable is used by consecutive iterations of
2012-12-19 15:22:24 +08:00
the loop. Normally, this would prevent vectorization, but the vectorizer can
2012-12-20 10:23:25 +08:00
detect that 'sum' is a reduction variable. The variable 'sum' becomes a vector
2012-12-19 15:22:24 +08:00
of integers, and at the end of the loop the elements of the array are added
2012-12-21 06:59:36 +08:00
together to create the correct result. We support a number of different
2012-12-19 15:22:24 +08:00
reduction operations, such as addition, multiplication, XOR, AND and OR.
.. code-block :: c++
int foo(int *A, int * B, int n) {
unsigned sum = 0;
for (int i = 0; i < n; ++i)
2012-12-21 06:59:36 +08:00
sum += A[i] + 5;
2012-12-19 15:22:24 +08:00
return sum;
}
2013-01-09 01:46:30 +08:00
We support floating point reduction operations when `-ffast-math` is used.
2012-12-19 15:22:24 +08:00
Inductions
2012-12-20 10:40:45 +08:00
^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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, float* B, float K, int n) {
2012-12-21 06:47:41 +08:00
for (int i = 0; i < n; ++i)
A[i] = i;
2012-12-19 15:22:24 +08:00
}
If Conversion
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
The Loop Vectorizer can vectorize loops that count backwards.
.. code-block :: c++
int foo(int *A, int * B, int n) {
for (int i = n; i > 0; --i)
A[i] +=1;
}
Scatter / Gather
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
2013-01-03 09:47:02 +08:00
The Loop Vectorizer can vectorize code that becomes a sequence of scalar instructions
that scatter/gathers memory.
2012-12-19 15:22:24 +08:00
.. code-block :: c++
int foo(int *A, int * B, int n, int k) {
2012-12-21 06:47:41 +08:00
for (int i = 0; i < n; ++i)
2012-12-21 06:49:13 +08:00
A[i*7] += B[i* k];
2012-12-19 15:22:24 +08:00
}
2012-12-19 15:36:35 +08:00
Vectorization of Mixed Types
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
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, int k) {
2012-12-21 06:47:41 +08:00
for (int i = 0; i < n; ++i)
2012-12-21 06:49:13 +08:00
A[i] += 4 * B[i];
2012-12-19 15:22:24 +08:00
}
Vectorization of function calls
2012-12-20 10:40:45 +08:00
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
2012-12-19 15:22:24 +08:00
The Loop Vectorize 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|
+-----+-----+---------+
2012-12-26 14:03:35 +08:00
| | | fmuladd |
+-----+-----+---------+
2012-12-19 15:22:24 +08:00
2013-01-03 09:47:02 +08:00
Partial unrolling during vectorization
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
Modern processors feature multiple execution units, and only programs that contain a
2013-01-03 09:56:33 +08:00
high degree of parallelism can fully utilize the entire width of the machine.
2013-01-03 09:47:02 +08:00
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'.
2013-01-03 09:56:33 +08:00
This is inefficient because only a single execution port can be used by the processor.
2013-01-03 09:47:02 +08:00
By unrolling the code the Loop Vectorizer allows two or more execution ports
2013-01-03 09:56:33 +08:00
to be used simultaneously.
2013-01-03 09:47:02 +08:00
.. code-block :: c++
int foo(int *A, int * B, int n) {
unsigned sum = 0;
for (int i = 0; i < n; ++i)
sum += A[i];
return sum;
}
2013-01-05 01:49:45 +08:00
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.
2013-01-03 09:47:02 +08:00
2012-12-19 16:28:24 +08:00
Performance
2012-12-20 10:40:45 +08:00
-----------
2012-12-19 16:28:24 +08:00
2012-12-21 06:59:36 +08:00
This section shows the the execution time of Clang on a simple benchmark:
2012-12-19 16:43:05 +08:00
`gcc-loops <http://llvm.org/viewvc/llvm-project/test-suite/trunk/SingleSource/UnitTests/Vectorizer/> `_ .
2012-12-21 06:59:36 +08:00
This benchmarks is a collection of loops from the GCC autovectorization
2012-12-20 02:02:36 +08:00
`page <http://gcc.gnu.org/projects/tree-ssa/vectorization.html> `_ by Dorit Nuzman.
2012-12-19 16:28:24 +08:00
2012-12-20 08:03:36 +08:00
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.
2012-12-21 06:59:36 +08:00
The Y-axis shows the time in msec. Lower is better. The last column shows the geomean of all the kernels.
2012-12-19 16:28:24 +08:00
.. image :: gcc-loops.png
2013-01-05 03:00:42 +08:00
And Linpack-pc with the same configuration. Result is Mflops, higher is better.
.. image :: linpack-pc.png
2012-12-21 06:42:20 +08:00
.. _bb-vectorizer:
2012-12-19 15:22:24 +08:00
The Basic Block Vectorizer
==========================
2012-12-20 02:04:44 +08:00
Usage
2012-12-20 10:40:45 +08:00
------
2012-12-20 02:04:44 +08:00
2012-12-19 15:22:24 +08:00
The Basic Block Vectorizer is not enabled by default, but it can be enabled
through clang using the command line flag:
.. code-block :: console
2012-12-21 06:59:36 +08:00
$ clang -fslp-vectorize file.c
2012-12-19 15:22:24 +08:00
2012-12-20 02:04:44 +08:00
Details
2012-12-20 10:40:45 +08:00
-------
2012-12-20 02:04:44 +08:00
2012-12-19 15:22:24 +08:00
The goal of basic-block vectorization (a.k.a. superword-level parallelism) is
to combine similar independent instructions within simple control-flow regions
into vector instructions. Memory accesses, arithemetic operations, comparison
operations and some math functions can all be vectorized using this technique
2012-12-21 06:59:36 +08:00
(subject to the capabilities of the target architecture).
2012-12-19 15:22:24 +08:00
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++
int foo(int a1, int a2, int b1, int b2) {
int r1 = a1*(a1 + b1)/b1 + 50* b1/a1;
int r2 = a2*(a2 + b2)/b2 + 50* b2/a2;
return r1 + r2;
}