Register Allocator Code

The register allocator code is split into two main sections, the register allocator proper and a generic graph coloring library. The graph coloring library is also used by the Stg->Cmm converter.

The register allocator

  • compiler/nativeGen/RegLiveness.hs
    Defines LiveInstr and LiveCmmTop which carry native machine instructions annotated with register liveness information. It also provides functions to annotate native code (NatCmmTop) with this liveness information, and to slurp out sets of register conflicts for feeding into the coloring allocator.
  • compiler/nativeGen/RegAllocColor.hs
    Defines regAlloc, the main driver function for the graph coloring allocator. The driver accepts LiveCmmTops which use virtual regs, and producesNatCmmTops which use real machine regs. This module also provides functions to help build and deep seq the register conflict graph.
  • compiler/nativeGen/RegAllocInfo.hs
    Defines the register information function, regUsage, which takes a set of real and virtual registers and returns the actual registers used by a particular Instr; register allocation is in AT&T syntax order (source, destination), in an internal function, usage; defines the RegUsage data type

  • compiler/nativeGen/RegSpill.hs
    Defines regSpill which takes LiveCmmTops and inserts spill/reload instructions virtual regs that wouldn't fit in real regs. regSpill's strategy is to simply inserts spill/reloads for every use/def of a particular virtual reg. This inefficient code is cleaned up by the spill cleaner after allocation.

  • compiler/nativeGen/RegSpillClean.hs
    The spill cleaner is run after real regs have been allocated. It erases spill/reload instructions inserted by regSpill that weren't strictly nessesary.

Graph coloring

  • compiler/utils/GraphColor.hs
    Defines the function colorGraph which is responsible for assigning colors (real regs) to nodes (virtual regs) in the register conflict graph.
  • compiler/utils/GraphOps.hs
    Defines functions to perform basic operations on the graphs such as adding, deleting, and coalescing nodes.


  • compiler/nativeGen/RegCoalesce.hs
    Defines a function regCoalesce that does aggressive coalescing directly on LiveCmmTops, without using the graph. This isn't used at the moment but has been left in incase we want to rejig the allocator when the new CPS converter comes online.
  • compiler/nativeGen/RegArchBase.hs
    Defines utils for calculating whether a register in the conflict graph is trivially colorable, in a generic way which handles aliasing between register classes. This module is not used directly by GHC.
Last modified 6 years ago Last modified on Jun 10, 2014 6:52:39 PM