Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 1 | ========================== |
| 2 | Auto-Vectorization in LLVM |
| 3 | ========================== |
| 4 | |
Sean Silva | 12ae515 | 2012-12-20 22:42:20 +0000 | [diff] [blame] | 5 | .. contents:: |
| 6 | :local: |
| 7 | |
| 8 | LLVM has two vectorizers: The :ref:`Loop Vectorizer <loop-vectorizer>`, |
| 9 | which operates on Loops, and the :ref:`Basic Block Vectorizer |
| 10 | <bb-vectorizer>`, which optimizes straight-line code. These vectorizers |
| 11 | focus on different optimization opportunities and use different techniques. |
| 12 | The BB vectorizer merges multiple scalars that are found in the code into |
| 13 | vectors while the Loop Vectorizer widens instructions in the original loop |
| 14 | to operate on multiple consecutive loop iterations. |
| 15 | |
| 16 | .. _loop-vectorizer: |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 17 | |
| 18 | The Loop Vectorizer |
| 19 | =================== |
| 20 | |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 21 | Usage |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 22 | ----- |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 23 | |
Sean Silva | 68d5b27 | 2012-12-20 02:23:25 +0000 | [diff] [blame] | 24 | LLVM's Loop Vectorizer is now available and will be useful for many people. |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 25 | It is not enabled by default, but can be enabled through clang using the |
| 26 | command line flag: |
| 27 | |
| 28 | .. code-block:: console |
| 29 | |
Nadav Rotem | 3e6da7e | 2012-12-19 18:02:36 +0000 | [diff] [blame] | 30 | $ clang -fvectorize -O3 file.c |
| 31 | |
| 32 | If the ``-fvectorize`` flag is used then the loop vectorizer will be enabled |
| 33 | when running with ``-O3``, ``-O2``. When ``-Os`` is used, the loop vectorizer |
| 34 | will only vectorize loops that do not require a major increase in code size. |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 35 | |
| 36 | We plan to enable the Loop Vectorizer by default as part of the LLVM 3.3 release. |
| 37 | |
Nadav Rotem | 4aa55bb | 2013-01-04 17:49:45 +0000 | [diff] [blame] | 38 | Command line flags |
| 39 | ^^^^^^^^^^^^^^^^^^ |
| 40 | |
| 41 | The loop vectorizer uses a cost model to decide on the optimal vectorization factor |
| 42 | and unroll factor. However, users of the vectorizer can force the vectorizer to use |
| 43 | specific values. Both 'clang' and 'opt' support the flags below. |
| 44 | |
| 45 | Users can control the vectorization SIMD width using the command line flag "-force-vector-width". |
| 46 | |
| 47 | .. code-block:: console |
| 48 | |
| 49 | $ clang -mllvm -force-vector-width=8 ... |
| 50 | $ opt -loop-vectorize -force-vector-width=8 ... |
| 51 | |
| 52 | Users can control the unroll factor using the command line flag "-force-vector-unroll" |
| 53 | |
| 54 | .. code-block:: console |
| 55 | |
| 56 | $ clang -mllvm -force-vector-unroll=2 ... |
| 57 | $ opt -loop-vectorize -force-vector-unroll=2 ... |
| 58 | |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 59 | Features |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 60 | -------- |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 61 | |
| 62 | The LLVM Loop Vectorizer has a number of features that allow it to vectorize |
| 63 | complex loops. |
| 64 | |
| 65 | Loops with unknown trip count |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 66 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 67 | |
| 68 | The Loop Vectorizer supports loops with an unknown trip count. |
| 69 | In the loop below, the iteration ``start`` and ``finish`` points are unknown, |
| 70 | and the Loop Vectorizer has a mechanism to vectorize loops that do not start |
Sean Silva | 68d5b27 | 2012-12-20 02:23:25 +0000 | [diff] [blame] | 71 | at zero. In this example, 'n' may not be a multiple of the vector width, and |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 72 | the vectorizer has to execute the last few iterations as scalar code. Keeping |
| 73 | a scalar copy of the loop increases the code size. |
| 74 | |
| 75 | .. code-block:: c++ |
| 76 | |
| 77 | void bar(float *A, float* B, float K, int start, int end) { |
Sean Silva | 9baa6e4 | 2012-12-20 22:47:41 +0000 | [diff] [blame] | 78 | for (int i = start; i < end; ++i) |
| 79 | A[i] *= B[i] + K; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 80 | } |
| 81 | |
| 82 | Runtime Checks of Pointers |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 83 | ^^^^^^^^^^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 84 | |
| 85 | In the example below, if the pointers A and B point to consecutive addresses, |
| 86 | then it is illegal to vectorize the code because some elements of A will be |
| 87 | written before they are read from array B. |
| 88 | |
| 89 | Some programmers use the 'restrict' keyword to notify the compiler that the |
| 90 | pointers are disjointed, but in our example, the Loop Vectorizer has no way of |
| 91 | knowing that the pointers A and B are unique. The Loop Vectorizer handles this |
| 92 | loop by placing code that checks, at runtime, if the arrays A and B point to |
| 93 | disjointed memory locations. If arrays A and B overlap, then the scalar version |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 94 | of the loop is executed. |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 95 | |
| 96 | .. code-block:: c++ |
| 97 | |
| 98 | void bar(float *A, float* B, float K, int n) { |
Sean Silva | 9baa6e4 | 2012-12-20 22:47:41 +0000 | [diff] [blame] | 99 | for (int i = 0; i < n; ++i) |
| 100 | A[i] *= B[i] + K; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 101 | } |
| 102 | |
| 103 | |
| 104 | Reductions |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 105 | ^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 106 | |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 107 | In this example the ``sum`` variable is used by consecutive iterations of |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 108 | the loop. Normally, this would prevent vectorization, but the vectorizer can |
Sean Silva | 68d5b27 | 2012-12-20 02:23:25 +0000 | [diff] [blame] | 109 | detect that 'sum' is a reduction variable. The variable 'sum' becomes a vector |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 110 | of integers, and at the end of the loop the elements of the array are added |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 111 | together to create the correct result. We support a number of different |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 112 | reduction operations, such as addition, multiplication, XOR, AND and OR. |
| 113 | |
| 114 | .. code-block:: c++ |
| 115 | |
| 116 | int foo(int *A, int *B, int n) { |
| 117 | unsigned sum = 0; |
| 118 | for (int i = 0; i < n; ++i) |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 119 | sum += A[i] + 5; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 120 | return sum; |
| 121 | } |
| 122 | |
Nadav Rotem | 2a92c10 | 2013-01-08 17:46:30 +0000 | [diff] [blame] | 123 | We support floating point reduction operations when `-ffast-math` is used. |
| 124 | |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 125 | Inductions |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 126 | ^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 127 | |
| 128 | In this example the value of the induction variable ``i`` is saved into an |
| 129 | array. The Loop Vectorizer knows to vectorize induction variables. |
| 130 | |
| 131 | .. code-block:: c++ |
| 132 | |
| 133 | void bar(float *A, float* B, float K, int n) { |
Sean Silva | 9baa6e4 | 2012-12-20 22:47:41 +0000 | [diff] [blame] | 134 | for (int i = 0; i < n; ++i) |
| 135 | A[i] = i; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 136 | } |
| 137 | |
| 138 | If Conversion |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 139 | ^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 140 | |
| 141 | The Loop Vectorizer is able to "flatten" the IF statement in the code and |
| 142 | generate a single stream of instructions. The Loop Vectorizer supports any |
| 143 | control flow in the innermost loop. The innermost loop may contain complex |
| 144 | nesting of IFs, ELSEs and even GOTOs. |
| 145 | |
| 146 | .. code-block:: c++ |
| 147 | |
| 148 | int foo(int *A, int *B, int n) { |
| 149 | unsigned sum = 0; |
| 150 | for (int i = 0; i < n; ++i) |
| 151 | if (A[i] > B[i]) |
| 152 | sum += A[i] + 5; |
| 153 | return sum; |
| 154 | } |
| 155 | |
| 156 | Pointer Induction Variables |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 157 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 158 | |
| 159 | This example uses the "accumulate" function of the standard c++ library. This |
| 160 | loop uses C++ iterators, which are pointers, and not integer indices. |
| 161 | The Loop Vectorizer detects pointer induction variables and can vectorize |
| 162 | this loop. This feature is important because many C++ programs use iterators. |
| 163 | |
| 164 | .. code-block:: c++ |
| 165 | |
| 166 | int baz(int *A, int n) { |
| 167 | return std::accumulate(A, A + n, 0); |
| 168 | } |
| 169 | |
| 170 | Reverse Iterators |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 171 | ^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 172 | |
| 173 | The Loop Vectorizer can vectorize loops that count backwards. |
| 174 | |
| 175 | .. code-block:: c++ |
| 176 | |
| 177 | int foo(int *A, int *B, int n) { |
| 178 | for (int i = n; i > 0; --i) |
| 179 | A[i] +=1; |
| 180 | } |
| 181 | |
| 182 | Scatter / Gather |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 183 | ^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 184 | |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 185 | The Loop Vectorizer can vectorize code that becomes a sequence of scalar instructions |
| 186 | that scatter/gathers memory. |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 187 | |
| 188 | .. code-block:: c++ |
| 189 | |
| 190 | int foo(int *A, int *B, int n, int k) { |
Sean Silva | 9baa6e4 | 2012-12-20 22:47:41 +0000 | [diff] [blame] | 191 | for (int i = 0; i < n; ++i) |
Sean Silva | 5e81633 | 2012-12-20 22:49:13 +0000 | [diff] [blame] | 192 | A[i*7] += B[i*k]; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 193 | } |
| 194 | |
Nadav Rotem | af08627 | 2012-12-19 07:36:35 +0000 | [diff] [blame] | 195 | Vectorization of Mixed Types |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 196 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 197 | |
| 198 | The Loop Vectorizer can vectorize programs with mixed types. The Vectorizer |
| 199 | cost model can estimate the cost of the type conversion and decide if |
| 200 | vectorization is profitable. |
| 201 | |
| 202 | .. code-block:: c++ |
| 203 | |
| 204 | int foo(int *A, char *B, int n, int k) { |
Sean Silva | 9baa6e4 | 2012-12-20 22:47:41 +0000 | [diff] [blame] | 205 | for (int i = 0; i < n; ++i) |
Sean Silva | 5e81633 | 2012-12-20 22:49:13 +0000 | [diff] [blame] | 206 | A[i] += 4 * B[i]; |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 207 | } |
| 208 | |
| 209 | Vectorization of function calls |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 210 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 211 | |
| 212 | The Loop Vectorize can vectorize intrinsic math functions. |
| 213 | See the table below for a list of these functions. |
| 214 | |
| 215 | +-----+-----+---------+ |
| 216 | | pow | exp | exp2 | |
| 217 | +-----+-----+---------+ |
| 218 | | sin | cos | sqrt | |
| 219 | +-----+-----+---------+ |
| 220 | | log |log2 | log10 | |
| 221 | +-----+-----+---------+ |
| 222 | |fabs |floor| ceil | |
| 223 | +-----+-----+---------+ |
| 224 | |fma |trunc|nearbyint| |
| 225 | +-----+-----+---------+ |
Nadav Rotem | f7769e3 | 2012-12-26 06:03:35 +0000 | [diff] [blame] | 226 | | | | fmuladd | |
| 227 | +-----+-----+---------+ |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 228 | |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 229 | |
| 230 | Partial unrolling during vectorization |
| 231 | ^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^ |
| 232 | |
| 233 | Modern processors feature multiple execution units, and only programs that contain a |
Nadav Rotem | 43f3928 | 2013-01-03 01:56:33 +0000 | [diff] [blame] | 234 | high degree of parallelism can fully utilize the entire width of the machine. |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 235 | The Loop Vectorizer increases the instruction level parallelism (ILP) by |
| 236 | performing partial-unrolling of loops. |
| 237 | |
| 238 | In the example below the entire array is accumulated into the variable 'sum'. |
Nadav Rotem | 43f3928 | 2013-01-03 01:56:33 +0000 | [diff] [blame] | 239 | This is inefficient because only a single execution port can be used by the processor. |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 240 | By unrolling the code the Loop Vectorizer allows two or more execution ports |
Nadav Rotem | 43f3928 | 2013-01-03 01:56:33 +0000 | [diff] [blame] | 241 | to be used simultaneously. |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 242 | |
| 243 | .. code-block:: c++ |
| 244 | |
| 245 | int foo(int *A, int *B, int n) { |
| 246 | unsigned sum = 0; |
| 247 | for (int i = 0; i < n; ++i) |
| 248 | sum += A[i]; |
| 249 | return sum; |
| 250 | } |
| 251 | |
Nadav Rotem | 4aa55bb | 2013-01-04 17:49:45 +0000 | [diff] [blame] | 252 | The Loop Vectorizer uses a cost model to decide when it is profitable to unroll loops. |
| 253 | The decision to unroll the loop depends on the register pressure and the generated code size. |
Nadav Rotem | f574b88 | 2013-01-03 01:47:02 +0000 | [diff] [blame] | 254 | |
Nadav Rotem | 67a6ec8 | 2012-12-19 08:28:24 +0000 | [diff] [blame] | 255 | Performance |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 256 | ----------- |
Nadav Rotem | 67a6ec8 | 2012-12-19 08:28:24 +0000 | [diff] [blame] | 257 | |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 258 | This section shows the the execution time of Clang on a simple benchmark: |
Nadav Rotem | 0575429 | 2012-12-19 08:43:05 +0000 | [diff] [blame] | 259 | `gcc-loops <http://llvm.org/viewvc/llvm-project/test-suite/trunk/SingleSource/UnitTests/Vectorizer/>`_. |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 260 | This benchmarks is a collection of loops from the GCC autovectorization |
Nadav Rotem | 3e6da7e | 2012-12-19 18:02:36 +0000 | [diff] [blame] | 261 | `page <http://gcc.gnu.org/projects/tree-ssa/vectorization.html>`_ by Dorit Nuzman. |
Nadav Rotem | 67a6ec8 | 2012-12-19 08:28:24 +0000 | [diff] [blame] | 262 | |
Nadav Rotem | 6d1fc53 | 2012-12-20 00:03:36 +0000 | [diff] [blame] | 263 | 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. |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 264 | The Y-axis shows the time in msec. Lower is better. The last column shows the geomean of all the kernels. |
Nadav Rotem | 67a6ec8 | 2012-12-19 08:28:24 +0000 | [diff] [blame] | 265 | |
| 266 | .. image:: gcc-loops.png |
| 267 | |
Nadav Rotem | 13410a1 | 2013-01-04 19:00:42 +0000 | [diff] [blame] | 268 | And Linpack-pc with the same configuration. Result is Mflops, higher is better. |
| 269 | |
| 270 | .. image:: linpack-pc.png |
| 271 | |
Sean Silva | 12ae515 | 2012-12-20 22:42:20 +0000 | [diff] [blame] | 272 | .. _bb-vectorizer: |
| 273 | |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 274 | The Basic Block Vectorizer |
| 275 | ========================== |
| 276 | |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 277 | Usage |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 278 | ------ |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 279 | |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 280 | The Basic Block Vectorizer is not enabled by default, but it can be enabled |
| 281 | through clang using the command line flag: |
| 282 | |
| 283 | .. code-block:: console |
| 284 | |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 285 | $ clang -fslp-vectorize file.c |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 286 | |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 287 | Details |
Sean Silva | 6241703 | 2012-12-20 02:40:45 +0000 | [diff] [blame] | 288 | ------- |
Nadav Rotem | 649a33e | 2012-12-19 18:04:44 +0000 | [diff] [blame] | 289 | |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 290 | The goal of basic-block vectorization (a.k.a. superword-level parallelism) is |
| 291 | to combine similar independent instructions within simple control-flow regions |
| 292 | into vector instructions. Memory accesses, arithemetic operations, comparison |
| 293 | operations and some math functions can all be vectorized using this technique |
Sean Silva | 689858b | 2012-12-20 22:59:36 +0000 | [diff] [blame] | 294 | (subject to the capabilities of the target architecture). |
Nadav Rotem | 59f2af9 | 2012-12-19 07:22:24 +0000 | [diff] [blame] | 295 | |
| 296 | For example, the following function performs very similar operations on its |
| 297 | inputs (a1, b1) and (a2, b2). The basic-block vectorizer may combine these |
| 298 | into vector operations. |
| 299 | |
| 300 | .. code-block:: c++ |
| 301 | |
| 302 | int foo(int a1, int a2, int b1, int b2) { |
| 303 | int r1 = a1*(a1 + b1)/b1 + 50*b1/a1; |
| 304 | int r2 = a2*(a2 + b2)/b2 + 50*b2/a2; |
| 305 | return r1 + r2; |
| 306 | } |
| 307 | |
| 308 | |