extra/compiler/cfg/gvn/gvn.factor

   1 ! Copyright (C) 2008, 2010 Slava Pestov.
   2 ! See http://factorcode.org/license.txt for BSD license.
   3 USING: namespaces arrays assocs kernel accessors fry grouping
   4 sorting sets sequences locals
   5 cpu.architecture
   6 sequences.deep
   7 compiler.cfg
   8 compiler.cfg.rpo
   9 compiler.cfg.def-use
  10 compiler.cfg.utilities
  11 compiler.cfg.instructions
  12 compiler.cfg.gvn.alien
  13 compiler.cfg.gvn.comparisons
  14 compiler.cfg.gvn.graph
  15 compiler.cfg.gvn.math
  16 compiler.cfg.gvn.rewrite
  17 compiler.cfg.gvn.slots
  18 compiler.cfg.gvn.misc
  19 compiler.cfg.gvn.expressions ;
  20 IN: compiler.cfg.gvn
  21
  22 GENERIC: process-instruction ( insn -- insn' )
  23
  24 : redundant-instruction ( insn vn -- insn' )
  25     [ dst>> ] dip [ swap set-vn ] [ <copy> ] 2bi ;
  26
  27 :: useful-instruction ( insn expr -- insn' )
  28     insn dst>> :> vn
  29     vn vn set-vn
  30     vn expr exprs>vns get set-at
  31     insn vn vns>insns get set-at
  32     insn ;
  33
  34 : check-redundancy ( insn -- insn' )
  35     dup >expr dup exprs>vns get at
  36     [ redundant-instruction ] [ useful-instruction ] ?if ;
  37
  38 M: insn process-instruction
  39     dup rewrite [ process-instruction ] [ ] ?if ;
  40
  41 M: foldable-insn process-instruction
  42     dup rewrite
  43     [ process-instruction ]
  44     [ dup defs-vregs length 1 = [ check-redundancy ] when ] ?if ;
  45
  46 M: ##copy process-instruction
  47     dup [ src>> vreg>vn ] [ dst>> ] bi set-vn ;
  48
  49 M: ##phi rewrite
  50     [ dst>> ] [ inputs>> values [ vreg>vn ] map ] bi
  51     dup sift
  52     dup all-equal?  [
  53         nip
  54         [ drop f ]
  55         [ first <copy> ] if-empty
  56     ] [ 3drop f ] if ;
  57
  58 M: ##phi process-instruction
  59     dup rewrite
  60     [ process-instruction ] [ check-redundancy ] ?if ;
  61
  62 M: ##phi >expr
  63     inputs>> values [ vreg>vn ] map \ ##phi prefix ;
  64
  65 M: array process-instruction
  66     [ process-instruction ] map ;
  67
  68 : value-numbering-step ( insns -- insns' )
  69     init-value-graph
  70     ! [ process-instruction ] map flatten ;
  71
  72     ! idea: let rewrite do the constant/copy propagation (as
  73     ! that eventually leads to better VNs), but don't actually
  74     ! use them here, since changing the CFG mid-optimistic-GVN
  75     ! won't be sound
  76     dup [ process-instruction drop ] each ;
  77
  78 : value-numbering-iteration ( cfg -- )
  79     [ value-numbering-step ] simple-optimization ;
  80
  81 : value-numbering ( cfg -- cfg )
  82     dup
  83     init-gvn
  84     '[
  85         changed? off
  86         _ value-numbering-iteration
  87         changed? get
  88     ] loop
  89
  90     dup [ init-value-graph [ process-instruction ] map flatten ] simple-optimization
  91     cfg-changed predecessors-changed ;