Home

New PBQP solver, and updates to the PBQP graph.

Press ? to show keyboard shortcuts.
Committed
lhamesFeb 28 2014, 2:25 PM
Parents
rL202550: [PECOFF] Sort SEH table entries according to its value.
Branches
Unknown
Tags
Unknown
Subscribers
None
Projects
None
Description

New PBQP solver, and updates to the PBQP graph.

The previous PBQP solver was very robust but consumed a lot of memory,
performed a lot of redundant computation, and contained some unnecessarily tight
coupling that prevented experimentation with novel solution techniques. This new
solver is an attempt to address these shortcomings.

Important/interesting changes:

  1. The domain-independent PBQP solver class, HeuristicSolverImpl, is gone.

It is replaced by a register allocation specific solver, PBQP::RegAlloc::Solver
(see RegAllocSolver.h).

The optimal reduction rules and the backpropagation algorithm have been extracted
into stand-alone functions (see ReductionRules.h), which can be used to build
domain specific PBQP solvers. This provides many more opportunities for
domain-specific knowledge to inform the PBQP solvers' decisions. In theory this
should allow us to generate better solutions. In practice, we can at least test
out ideas now.

As a side benefit, I believe the new solver is more readable than the old one.

  1. The solver type is now a template parameter of the PBQP graph.

This allows the graph to notify the solver of any modifications made (e.g. by
domain independent rules) without the overhead of a virtual call. It also allows
the solver to supply policy information to the graph (see below).

  1. Significantly reduced memory overhead.

Memory management policy is now an explicit property of the PBQP graph (via
the CostAllocator typedef on the graph's solver template argument). Because PBQP
graphs for register allocation tend to contain many redundant instances of
single values (E.g. the value representing an interference constraint between
GPRs), the new RASolver class uses a uniquing scheme. This massively reduces
memory consumption for large register allocation problems. For example, looking
at the largest interference graph in each of the SPEC2006 benchmarks (the
largest graph will always set the memory consumption high-water mark for PBQP),
the average memory reduction for the PBQP costs was 400x. That's times, not
percent. The highest was 1400x. Yikes. So - this is fixed.

"PBQP: No longer feasting upon every last byte of your RAM".

Minor details:

  • Fully C++11'd. Never copy-construct another vector/matrix!
  • Cute tricks with cost metadata: Metadata that is derived solely from cost

matrices/vectors is attached directly to the cost instances themselves. That way
if you unique the costs you never have to recompute the metadata. 400x less
memory means 400x less cost metadata (re)computation.

Special thanks to Arnaud de Grandmaison, who has been the source of much
encouragement, and of many very useful test cases.

This new solver forms the basis for future work, of which there's plenty to do.
I will be adding TODO notes shortly.

  • Lang.

rL202551

llvm/trunk/include/llvm/CodeGen/PBQP/CostAllocator.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/Graph.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/HeuristicBase.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/HeuristicSolver.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/Heuristics/

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/Heuristics/Briggs.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/Math.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/ReductionRules.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/RegAllocSolver.h

Loading...

llvm/trunk/include/llvm/CodeGen/PBQP/Solution.h

Loading...

llvm/trunk/include/llvm/CodeGen/RegAllocPBQP.h

Loading...

llvm/trunk/lib/CodeGen/RegAllocPBQP.cpp

Loading...

Add Comment