Loop fission and fusion explained

In computer science, loop fission (or loop distribution) is a compiler optimization in which a loop is broken into multiple loops over the same index range with each taking only a part of the original loop's body.[1] [2] The goal is to break down a large loop body into smaller ones to achieve better utilization of locality of reference. This optimization is most efficient in multi-core processors that can split a task into multiple tasks for each processor.

Conversely, loop fusion (or loop jamming) is a compiler optimization and loop transformation which replaces multiple loops with a single one.[3] [2] Loop fusion does not always improve run-time speed. On some architectures, two loops may actually perform better than one loop because, for example, there is increased data locality within each loop. One of the main benefits of loop fusion is that it allows temporary allocations to be avoided, which can lead to huge performance gains in numerical computing languages such as Julia when doing elementwise operations on arrays (however, Julia's loop fusion is not technically a compiler optimization, but a syntactic guarantee of the language).[4]

Other benefits of loop fusion are that it avoids the overhead of the loop control structures, and also that it allows the loop body to be parallelized by the processor[5] by taking advantage of instruction-level parallelism. This is possible when there are no data dependencies between the bodies of the two loops (this is in stark contrast to the other main benefit of loop fusion described above, which only presents itself when there are data dependencies that require an intermediate allocation to store the results). If loop fusion is able to remove redundant allocations, performance increases can be large. Otherwise, there is a more complex trade-off between data locality, instruction-level parallelism, and loop overhead (branching, incrementing, etc.) that may make loop fusion, loop fission, or neither, the preferable optimization.

Fission

Example in C

int i, a[100], b[100];for (i = 0; i < 100; i++)

is equivalent to:

int i, a[100], b[100];for (i = 0; i < 100; i++) for (i = 0; i < 100; i++)

Fusion

Example in C++ and MATLAB

Consider the following MATLAB code:x = 0:999; % Create an array of numbers from 0 to 999 (range is inclusive)y = sin(x) + 4; % Take the sine of x (element-wise) and add 4 to each element

You could achieve the same syntax in C++ by using function and operator overloading:

  1. include
  2. include
  3. include
  4. include

class Array ;

int main(int argc, char* argv[]) However, the above example unnecessarily allocates a temporary array for the result of sin(x). A more efficient implementation would allocate a single array for y, and compute y in a single loop. To optimize this, a C++ compiler would need to:

  1. Inline the sin and operator+ function calls.
  2. Fuse the loops into a single loop.
  3. Remove the unused stores into the temporary arrays (can use a register or stack variable instead).
  4. Remove the unused allocation and free.

All of these steps are individually possible. Even step four is possible despite the fact that functions like malloc and free have global side effects, since some compilers hardcode symbols such as malloc and free so that they can remove unused allocations from the code.[6] However, as of clang 12.0.0 and gcc 11.1, this loop fusion and redundant allocation removal does not occur - even on the highest optimization level.[7] [8]

Some languages specifically targeted towards numerical computing such as Julia might have the concept of loop fusion built into it at a high level, where the compiler will notice adjacent elementwise operations and fuse them into a single loop.[9] Currently, to achieve the same syntax in general purpose languages like C++, the sin and operator+ functions must pessimistically allocate arrays to store their results, since they do not know what context they will be called from. This issue can be avoided in C++ by using a different syntax that does not rely on the compiler to remove unnecessary temporary allocations (e.g., using functions and overloads for in-place operations, such as operator+= or std::transform).

See also

References

Notes and References

  1. Book: Y.N. Srikant. Priti Shankar. The Compiler Design Handbook: Optimizations and Machine Code Generation, Second Edition. 3 October 2018. CRC Press. 978-1-4200-4383-9.
  2. Book: Kennedy, Ken . Allen, Randy. . amp . Optimizing Compilers for Modern Architectures: A Dependence-based Approach . Morgan Kaufmann . 2001 . 1-55860-286-0.
  3. Book: Steven Muchnick. Muchnick and Associates. Advanced Compiler Design Implementation. registration. loop fusion.. 15 August 1997. Morgan Kaufmann. 978-1-55860-320-2.
  4. Web site: Johnson. Steven G.. 21 January 2017. More Dots: Syntactic Loop Fusion in Julia. 25 June 2021. julialang.org.
  5. Web site: Loop Fusion. 2021-06-25. Intel. en.
  6. Web site: Godbolt. Matt. Compiler Explorer - C++ (x86-64 clang 12.0.0). 2021-06-25. godbolt.org. en.
  7. Web site: Godbolt. Matt. Compiler Explorer - C++ (x86-64 clang 12.0.0). 2021-06-25. godbolt.org. en.
  8. Web site: Godbolt. Matt. Compiler Explorer - C++ (x86-64 gcc 11.1). 2021-06-25. godbolt.org. en.
  9. Web site: Functions ยท The Julia Language. 2021-06-25. docs.julialang.org.