basic-block get successors>> push
stop-iterating ;
-: emit-call ( word -- next )
+: emit-call ( word height -- next )
{
- { [ dup loops get key? ] [ loops get at local-recursive-call ] }
+ { [ over loops get key? ] [ drop loops get at local-recursive-call ] }
{ [ tail-call? not ] [ ##call ##branch begin-basic-block iterate-next ] }
- { [ dup current-label get eq? ] [ drop first-basic-block get local-recursive-call ] }
- [ ##epilogue ##jump stop-iterating ]
+ { [ dup current-label get eq? ] [ 2drop first-basic-block get local-recursive-call ] }
+ [ drop ##epilogue ##jump stop-iterating ]
} cond ;
! #recursive
-: compile-recursive ( node -- next )
- [ label>> id>> emit-call ]
+: recursive-height ( #recursive -- n )
+ [ label>> return>> in-d>> length ] [ in-d>> length ] bi - ;
+
+: emit-recursive ( #recursive -- next )
+ [ [ label>> id>> ] [ recursive-height ] bi emit-call ]
[ [ child>> ] [ label>> word>> ] [ label>> id>> ] tri (build-cfg) ] bi ;
: remember-loop ( label -- )
basic-block get swap loops get set-at ;
-: compile-loop ( node -- next )
+: emit-loop ( node -- next )
##loop-entry
begin-basic-block
[ label>> id>> remember-loop ] [ child>> emit-nodes ] bi
iterate-next ;
M: #recursive emit-node
- dup label>> loop?>> [ compile-loop ] [ compile-recursive ] if ;
+ dup label>> loop?>> [ emit-loop ] [ emit-recursive ] if ;
! #if
: emit-branch ( obj -- final-bb )
ds-pop ^^offset>slot i 0 ##dispatch
dispatch-branches ;
-: <dispatch-block> ( -- word )
+! If a dispatch is not in tail position, we compile a new word where the dispatch is in
+! tail position, then call this word.
+
+: (non-tail-dispatch) ( -- word )
gensym dup t "inlined-block" set-word-prop ;
+: <non-tail-dispatch> ( node -- word )
+ current-word get (non-tail-dispatch) [
+ [
+ begin-word
+ emit-dispatch
+ ] with-cfg-builder
+ ] keep ;
+
M: #dispatch emit-node
tail-call? [
emit-dispatch stop-iterating
] [
- current-word get <dispatch-block> [
- [
- begin-word
- emit-dispatch
- ] with-cfg-builder
- ] keep emit-call
+ <non-tail-dispatch> f emit-call
] if ;
! #call
M: #call emit-node
dup word>> dup "intrinsic" word-prop
- [ emit-intrinsic ] [ nip emit-call ] if ;
+ [ emit-intrinsic ] [ swap call-height emit-call ] if ;
! #call-recursive
-M: #call-recursive emit-node label>> id>> emit-call ;
+M: #call-recursive emit-node [ label>> id>> ] [ call-height ] bi emit-call ;
! #push
M: #push emit-node
--- /dev/null
+! Copyright (C) 2009 Slava Pestov.
+! See http://factorcode.org/license.txt for BSD license.
+USING: accessors assocs kernel namespaces math sequences fry deques grouping
+search-deques dlists sets make combinators compiler.cfg.copy-prop
+compiler.cfg.def-use compiler.cfg.instructions compiler.cfg.registers
+compiler.cfg.rpo compiler.cfg.hats ;
+IN: compiler.cfg.stack-analysis
+
+! Convert stack operations to register operations
+
+! If 'poisoned' is set, disregard height information. This is set if we don't have
+! height change information for an instruction.
+TUPLE: state locs>vregs vregs>locs changed-locs d-height r-height poisoned? ;
+
+: <state> ( -- state )
+ state new
+ H{ } clone >>locs>vregs
+ H{ } clone >>vregs>locs
+ H{ } clone >>changed-locs
+ 0 >>d-height
+ 0 >>r-height ;
+
+M: state clone
+ call-next-method
+ [ clone ] change-locs>vregs
+ [ clone ] change-vregs>locs
+ [ clone ] change-changed-locs ;
+
+: loc>vreg ( loc -- vreg ) state get locs>vregs>> at ;
+
+: record-peek ( dst loc -- )
+ state get
+ [ locs>vregs>> set-at ]
+ [ swapd vregs>locs>> set-at ]
+ 3bi ;
+
+: delete-old-vreg ( loc -- )
+ state get locs>vregs>> at [ state get vregs>locs>> delete-at ] when* ;
+
+: changed-loc ( loc -- )
+ state get changed-locs>> conjoin ;
+
+: redundant-replace? ( src loc -- ? )
+ loc>vreg = ;
+
+: record-replace ( src loc -- )
+ ! Locs are not single assignment, which means we have to forget
+ ! that the previous vreg, if any, points at this loc. Also, record
+ ! that the loc changed so that all the right ##replace instructions
+ ! are emitted at a sync point.
+ 2dup redundant-replace? [ 2drop ] [
+ dup delete-old-vreg dup changed-loc record-peek
+ ] if ;
+
+: save-changed-locs ( state -- )
+ [ changed-locs>> ] [ locs>vregs>> ] bi '[
+ _ at swap 2dup redundant-replace?
+ [ 2drop ] [ ##replace ] if
+ ] assoc-each ;
+
+: clear-state ( state -- )
+ {
+ [ 0 >>d-height drop ]
+ [ 0 >>r-height drop ]
+ [ changed-locs>> clear-assoc ]
+ [ locs>vregs>> clear-assoc ]
+ [ vregs>locs>> clear-assoc ]
+ } cleave ;
+
+ERROR: poisoned-state state ;
+
+: sync-state ( -- )
+ state get {
+ [ dup poisoned?>> [ poisoned-state ] [ drop ] if ]
+ [ save-changed-locs ]
+ [ d-height>> dup 0 = [ drop ] [ ##inc-d ] if ]
+ [ r-height>> dup 0 = [ drop ] [ ##inc-r ] if ]
+ [ clear-state ]
+ } cleave ;
+
+: poison-state ( -- ) state get t >>poisoned? drop ;
+
+GENERIC: translate-loc ( loc -- loc' )
+
+M: ds-loc translate-loc n>> state get d-height>> + <ds-loc> ;
+
+M: rs-loc translate-loc n>> state get r-height>> + <rs-loc> ;
+
+! Abstract interpretation
+GENERIC: visit ( insn -- )
+
+! Instructions which don't have any effect on the stack
+UNION: neutral-insn
+ ##flushable
+ ##effect
+ ##branch
+ ##loop-entry
+ ##conditional-branch ;
+
+M: neutral-insn visit , ;
+
+: adjust-d ( n -- ) state get [ + ] change-d-height drop ;
+
+M: ##inc-d visit n>> adjust-d ;
+
+: adjust-r ( n -- ) state get [ + ] change-r-height drop ;
+
+M: ##inc-r visit n>> adjust-r ;
+
+: eliminate-peek ( dst src -- )
+ ! the requested stack location is already in 'src'
+ [ ##copy ] [ swap copies get set-at ] 2bi ;
+
+M: ##peek visit
+ dup
+ [ dst>> ] [ loc>> translate-loc ] bi
+ dup loc>vreg dup [ nip eliminate-peek drop ] [ drop record-peek , ] if ;
+
+M: ##replace visit
+ [ src>> resolve ] [ loc>> translate-loc ] bi
+ record-replace ;
+
+M: ##copy visit
+ [ call-next-method ] [ record-copy ] bi ;
+
+M: ##call visit
+ [ call-next-method ] [ height>> [ adjust-d ] [ poison-state ] if* ] bi ;
+
+M: ##fixnum-mul visit
+ call-next-method -1 adjust-d ;
+
+M: ##fixnum-add visit
+ call-next-method -1 adjust-d ;
+
+M: ##fixnum-sub visit
+ call-next-method -1 adjust-d ;
+
+! Instructions that poison the stack state
+UNION: poison-insn
+ ##jump
+ ##return
+ ##dispatch
+ ##dispatch-label
+ ##alien-callback
+ ##callback-return
+ ##fixnum-mul-tail
+ ##fixnum-add-tail
+ ##fixnum-sub-tail ;
+
+M: poison-insn visit call-next-method poison-state ;
+
+! Instructions that kill all live vregs
+UNION: kill-vreg-insn
+ poison-insn
+ ##stack-frame
+ ##call
+ ##prologue
+ ##epilogue
+ ##fixnum-mul
+ ##fixnum-add
+ ##fixnum-sub
+ ##alien-invoke
+ ##alien-indirect ;
+
+M: kill-vreg-insn visit sync-state , ;
+
+: visit-alien-node ( node -- )
+ params>> [ out-d>> length ] [ in-d>> length ] bi - adjust-d ;
+
+M: ##alien-invoke visit
+ [ call-next-method ] [ visit-alien-node ] bi ;
+
+M: ##alien-indirect visit
+ [ call-next-method ] [ visit-alien-node ] bi ;
+
+! Basic blocks we still need to look at
+SYMBOL: work-list
+
+: add-to-work-list ( basic-block -- )
+ work-list get push-front ;
+
+! Maps basic-blocks to states
+SYMBOLS: state-in state-out ;
+
+: sync-unpoisoned-states ( predecessors states -- )
+ [
+ dup poisoned?>> [ 2drop ] [
+ state [
+ instructions>> building set
+ sync-state
+ ] with-variable
+ ] if
+ ] 2each ;
+
+ERROR: must-equal-failed seq ;
+
+: must-equal ( seq -- elt )
+ dup all-equal? [ first ] [ must-equal-failed ] if ;
+
+: merge-heights ( state predecessors states -- state )
+ nip
+ [ [ d-height>> ] map must-equal >>d-height ]
+ [ [ r-height>> ] map must-equal >>r-height ] bi ;
+
+ERROR: inconsistent-vreg>loc states ;
+
+: check-vreg>loc ( states -- )
+ ! The same vreg should not store different locs in
+ ! different branches
+ dup
+ [ vregs>locs>> ] map
+ [ [ keys ] map concat prune ] keep
+ '[ _ [ at ] with map sift all-equal? ] all?
+ [ drop ] [ inconsistent-vreg>loc ] if ;
+
+: insert-peek ( predecessor loc -- vreg )
+ ! XXX critical edges
+ [ instructions>> building ] dip '[ _ ^^peek ] with-variable ;
+
+: merge-loc ( predecessors locs>vregs loc -- vreg )
+ ! Insert a ##phi in the current block where the input
+ ! is the vreg storing loc from each predecessor block
+ [ '[ [ _ ] dip at ] map ] keep
+ '[ [ ] [ _ insert-peek ] if ] 2map
+ ^^phi ;
+
+: merge-locs ( state predecessors states -- state )
+ [ locs>vregs>> ] map dup [ keys ] map prune
+ [
+ [ 2nip ] [ merge-loc ] 3bi
+ ] with with H{ } map>assoc
+ >>locs>vregs ;
+
+: merge-states ( predecessors states -- state )
+ ! If any states are poisoned, save all registers
+ ! to the stack in each branch
+ [ drop <state> ] [
+ dup [ poisoned?>> ] any? [
+ sync-unpoisoned-states <state>
+ ] [
+ dup check-vreg>loc
+ [ state new ] 2dip
+ [ merge-heights ]
+ [ merge-locs ] 2bi
+ ! what about vregs>locs
+ ] if
+ ] if-empty ;
+
+: block-in-state ( bb -- states )
+ predecessors>> dup state-out get '[ _ at ] map merge-states ;
+
+: maybe-set-at ( value key assoc -- changed? )
+ 3dup at* [ = [ 3drop f ] [ set-at t ] if ] [ 2drop set-at t ] if ;
+
+: set-block-in-state ( state b -- )
+ state-in get set-at ;
+
+: set-block-out-state ( bb state -- changed? )
+ swap state-out get maybe-set-at ;
+
+: finish-block ( bb state -- )
+ [ drop ] [ set-block-out-state ] 2bi
+ [ successors>> [ add-to-work-list ] each ] [ drop ] if ;
+
+: visit-block ( bb -- )
+ ! block-in-state may add phi nodes at the start of the basic block
+ ! so we wrap the whole thing with a 'make'
+ [
+ dup block-in-state
+ [ swap set-block-in-state ] [
+ state [
+ [ instructions>> [ visit ] each ]
+ [ state get finish-block ]
+ [ ]
+ tri
+ ] with-variable
+ ] 2bi
+ ] V{ } make >>instructions drop ;
+
+: visit-blocks ( bb -- )
+ reverse-post-order work-list get
+ [ '[ _ push-front ] each ] [ [ visit-block ] slurp-deque ] bi ;
+
+: optimize-stack ( cfg -- cfg )
+ [
+ H{ } clone copies set
+ H{ } clone state-in set
+ H{ } clone state-out set
+ <hashed-dlist> work-list set
+ dup entry>> visit-blocks
+ ] with-scope ;
+
+! XXX: what if our height doesn't match
+! a future block we're merging with?
+! - we should only poison tail calls
+! - non-tail poisoning nodes: ##alien-callback, ##call of a non-tail dispatch
+! do we need a distinction between height changes in code and height changes done by the callee
\ No newline at end of file