1 unstable release

0.3.0 Oct 28, 2024

#860 in Algorithms

Download history 188/week @ 2024-10-28 62/week @ 2024-11-04 30/week @ 2024-11-11 26/week @ 2024-11-18 25/week @ 2024-11-25 15/week @ 2024-12-02 39/week @ 2024-12-09

106 downloads per month
Used in 6 crates (via cubecl-spirv)

MIT/Apache

675KB
17K SLoC

Compiler Optimizations

WIP


lib.rs:

CubeCL Optimizer

A library that parses CubeCL IR into a control flow graph, transforms it to static single-assignment form and runs various optimizations on it. The order of operations is as follows:

  1. Parse root scope recursively into a control flow graph
  2. Run optimizations that must be done before SSA transformation
  3. Analyze variable liveness
  4. Transform the graph to pruned SSA form
  5. Run post-SSA optimizations and analyses in a loop until no more improvements are found
  6. Speed

The output is represented as a petgraph graph of BasicBlocks terminated by ControlFlow. This can then be compiled into actual executable code by walking the graph and generating all phi nodes, instructions and branches.

Representing PhiInstruction in non-SSA languages

Phi instructions can be simulated by generating a mutable variable for each phi, then assigning value to it in each relevant block.

Dependencies

~6–20MB
~218K SLoC