compiler.cfg.gvn: merge in applicable changes from compiler.cfg.value-numbering

db4
Alex Vondrak 2012-08-02 16:41:23 -07:00 committed by John Benediktsson
parent 2456f90eaf
commit f2638c99dd
7 changed files with 88 additions and 88 deletions

View File

@ -34,7 +34,7 @@ M: ##box-displaced-alien rewrite
[ dst>> ] [ dst>> ]
[ [ base>> ] [ base-class>> ] [ displacement>> ] tri ] bi* [ [ base>> ] [ base-class>> ] [ displacement>> ] tri ] bi*
[ ^^unbox-c-ptr ] dip [ ^^unbox-c-ptr ] dip
##add ##add,
] { } make ; ] { } make ;
: rewrite-unbox-any-c-ptr ( insn -- insn/f ) : rewrite-unbox-any-c-ptr ( insn -- insn/f )
@ -99,8 +99,8 @@ M: ##store-memory-imm alien-insn-value src>> ;
GENERIC: new-alien-insn ( value base displacement scale offset rep c-type insn -- insn ) GENERIC: new-alien-insn ( value base displacement scale offset rep c-type insn -- insn )
M: ##load-memory-imm new-alien-insn drop \ ##load-memory new-insn ; M: ##load-memory-imm new-alien-insn drop ##load-memory new-insn ;
M: ##store-memory-imm new-alien-insn drop \ ##store-memory new-insn ; M: ##store-memory-imm new-alien-insn drop ##store-memory new-insn ;
: fuse-displacement ( insn -- insn' ) : fuse-displacement ( insn -- insn' )
{ {

View File

@ -87,15 +87,15 @@ UNION: general-compare-insn scalar-compare-insn ##test-vector ;
: rewrite-boolean-comparison ( insn -- insn ) : rewrite-boolean-comparison ( insn -- insn )
src1>> vreg>insn { src1>> vreg>insn {
{ [ dup ##compare? ] [ >compare< \ ##compare-branch new-insn ] } { [ dup ##compare? ] [ >compare< ##compare-branch new-insn ] }
{ [ dup ##compare-imm? ] [ >compare< \ ##compare-imm-branch new-insn ] } { [ dup ##compare-imm? ] [ >compare< ##compare-imm-branch new-insn ] }
{ [ dup ##compare-integer? ] [ >compare< \ ##compare-integer-branch new-insn ] } { [ dup ##compare-integer? ] [ >compare< ##compare-integer-branch new-insn ] }
{ [ dup ##compare-integer-imm? ] [ >compare< \ ##compare-integer-imm-branch new-insn ] } { [ dup ##compare-integer-imm? ] [ >compare< ##compare-integer-imm-branch new-insn ] }
{ [ dup ##test? ] [ >compare< \ ##test-branch new-insn ] } { [ dup ##test? ] [ >compare< ##test-branch new-insn ] }
{ [ dup ##test-imm? ] [ >compare< \ ##test-imm-branch new-insn ] } { [ dup ##test-imm? ] [ >compare< ##test-imm-branch new-insn ] }
{ [ dup ##compare-float-unordered? ] [ >compare< \ ##compare-float-unordered-branch new-insn ] } { [ dup ##compare-float-unordered? ] [ >compare< ##compare-float-unordered-branch new-insn ] }
{ [ dup ##compare-float-ordered? ] [ >compare< \ ##compare-float-ordered-branch new-insn ] } { [ dup ##compare-float-ordered? ] [ >compare< ##compare-float-ordered-branch new-insn ] }
{ [ dup ##test-vector? ] [ >test-vector< \ ##test-vector-branch new-insn ] } { [ dup ##test-vector? ] [ >test-vector< ##test-vector-branch new-insn ] }
} cond ; } cond ;
: fold-branch ( ? -- insn ) : fold-branch ( ? -- insn )
@ -103,13 +103,13 @@ UNION: general-compare-insn scalar-compare-insn ##test-vector ;
0 1 ? 0 1 ?
basic-block get [ nth 1vector ] change-successors drop basic-block get [ nth 1vector ] change-successors drop
] [ drop ] if ] [ drop ] if
\ ##branch new-insn ; ##branch new-insn ;
: fold-compare-imm-branch ( insn -- insn/f ) : fold-compare-imm-branch ( insn -- insn/f )
evaluate-compare-imm fold-branch ; evaluate-compare-imm fold-branch ;
: >test-branch ( insn -- insn ) : >test-branch ( insn -- insn )
[ src1>> ] [ src1>> ] [ cc>> ] tri \ ##test-branch new-insn ; [ src1>> ] [ src1>> ] [ cc>> ] tri ##test-branch new-insn ;
M: ##compare-imm-branch rewrite M: ##compare-imm-branch rewrite
{ {
@ -146,12 +146,12 @@ M: ##test-imm-branch rewrite
: >compare-imm-branch ( insn swap? -- insn' ) : >compare-imm-branch ( insn swap? -- insn' )
(>compare-imm-branch) (>compare-imm-branch)
[ vreg>literal ] dip [ vreg>literal ] dip
\ ##compare-imm-branch new-insn ; inline ##compare-imm-branch new-insn ; inline
: >compare-integer-imm-branch ( insn swap? -- insn' ) : >compare-integer-imm-branch ( insn swap? -- insn' )
(>compare-imm-branch) (>compare-imm-branch)
[ vreg>integer ] dip [ vreg>integer ] dip
\ ##compare-integer-imm-branch new-insn ; inline ##compare-integer-imm-branch new-insn ; inline
: evaluate-self-compare ( insn -- ? ) : evaluate-self-compare ( insn -- ? )
cc>> { cc= cc<= cc>= } member-eq? ; cc>> { cc= cc<= cc>= } member-eq? ;
@ -182,15 +182,15 @@ M: ##compare-integer-branch rewrite
: >compare-imm ( insn swap? -- insn' ) : >compare-imm ( insn swap? -- insn' )
(>compare-imm) (>compare-imm)
[ vreg>literal ] dip [ vreg>literal ] dip
next-vreg \ ##compare-imm new-insn ; inline next-vreg ##compare-imm new-insn ; inline
: >compare-integer-imm ( insn swap? -- insn' ) : >compare-integer-imm ( insn swap? -- insn' )
(>compare-imm) (>compare-imm)
[ vreg>integer ] dip [ vreg>integer ] dip
next-vreg \ ##compare-integer-imm new-insn ; inline next-vreg ##compare-integer-imm new-insn ; inline
: >boolean-insn ( insn ? -- insn' ) : >boolean-insn ( insn ? -- insn' )
[ dst>> ] dip \ ##load-reference new-insn ; [ dst>> ] dip ##load-reference new-insn ;
: rewrite-self-compare ( insn -- insn' ) : rewrite-self-compare ( insn -- insn' )
dup evaluate-self-compare >boolean-insn ; dup evaluate-self-compare >boolean-insn ;
@ -220,14 +220,14 @@ M: ##compare-integer rewrite
: rewrite-redundant-comparison ( insn -- insn' ) : rewrite-redundant-comparison ( insn -- insn' )
[ cc>> ] [ dst>> ] [ src1>> vreg>insn ] tri { [ cc>> ] [ dst>> ] [ src1>> vreg>insn ] tri {
{ [ dup ##compare? ] [ >compare< next-vreg \ ##compare new-insn ] } { [ dup ##compare? ] [ >compare< next-vreg ##compare new-insn ] }
{ [ dup ##compare-imm? ] [ >compare< next-vreg \ ##compare-imm new-insn ] } { [ dup ##compare-imm? ] [ >compare< next-vreg ##compare-imm new-insn ] }
{ [ dup ##compare-integer? ] [ >compare< next-vreg \ ##compare-integer new-insn ] } { [ dup ##compare-integer? ] [ >compare< next-vreg ##compare-integer new-insn ] }
{ [ dup ##compare-integer-imm? ] [ >compare< next-vreg \ ##compare-integer-imm new-insn ] } { [ dup ##compare-integer-imm? ] [ >compare< next-vreg ##compare-integer-imm new-insn ] }
{ [ dup ##test? ] [ >compare< next-vreg \ ##test new-insn ] } { [ dup ##test? ] [ >compare< next-vreg ##test new-insn ] }
{ [ dup ##test-imm? ] [ >compare< next-vreg \ ##test-imm new-insn ] } { [ dup ##test-imm? ] [ >compare< next-vreg ##test-imm new-insn ] }
{ [ dup ##compare-float-unordered? ] [ >compare< next-vreg \ ##compare-float-unordered new-insn ] } { [ dup ##compare-float-unordered? ] [ >compare< next-vreg ##compare-float-unordered new-insn ] }
{ [ dup ##compare-float-ordered? ] [ >compare< next-vreg \ ##compare-float-ordered new-insn ] } { [ dup ##compare-float-ordered? ] [ >compare< next-vreg ##compare-float-ordered new-insn ] }
} cond } cond
swap cc= eq? [ [ negate-cc ] change-cc ] when ; swap cc= eq? [ [ negate-cc ] change-cc ] when ;
@ -246,7 +246,7 @@ M: ##compare-imm rewrite
: >test ( insn -- insn' ) : >test ( insn -- insn' )
{ [ dst>> ] [ src1>> ] [ src1>> ] [ cc>> ] [ temp>> ] } cleave { [ dst>> ] [ src1>> ] [ src1>> ] [ cc>> ] [ temp>> ] } cleave
\ ##test new-insn ; ##test new-insn ;
M: ##compare-integer-imm rewrite M: ##compare-integer-imm rewrite
{ {
@ -262,10 +262,10 @@ M: ##compare-integer-imm rewrite
[ src1>> vreg>insn [ src1>> ] [ src2>> ] bi ] [ cc>> ] bi ; inline [ src1>> vreg>insn [ src1>> ] [ src2>> ] bi ] [ cc>> ] bi ; inline
: simplify-test ( insn -- insn ) : simplify-test ( insn -- insn )
[ dst>> ] [ (simplify-test) ] [ temp>> ] tri \ ##test new-insn ; inline [ dst>> ] [ (simplify-test) ] [ temp>> ] tri ##test new-insn ; inline
: simplify-test-branch ( insn -- insn ) : simplify-test-branch ( insn -- insn )
(simplify-test) \ ##test-branch new-insn ; inline (simplify-test) ##test-branch new-insn ; inline
: simplify-test-imm? ( insn -- ? ) : simplify-test-imm? ( insn -- ? )
src1>> vreg>insn [ ##and-imm? ] with-available-uses? ; src1>> vreg>insn [ ##and-imm? ] with-available-uses? ;
@ -274,18 +274,18 @@ M: ##compare-integer-imm rewrite
[ src1>> vreg>insn [ src1>> ] [ src2>> ] bi ] [ cc>> ] bi ; inline [ src1>> vreg>insn [ src1>> ] [ src2>> ] bi ] [ cc>> ] bi ; inline
: simplify-test-imm ( insn -- insn ) : simplify-test-imm ( insn -- insn )
[ dst>> ] [ (simplify-test-imm) ] [ temp>> ] tri \ ##test-imm new-insn ; inline [ dst>> ] [ (simplify-test-imm) ] [ temp>> ] tri ##test-imm new-insn ; inline
: simplify-test-imm-branch ( insn -- insn ) : simplify-test-imm-branch ( insn -- insn )
(simplify-test-imm) \ ##test-imm-branch new-insn ; inline (simplify-test-imm) ##test-imm-branch new-insn ; inline
: >test-imm ( insn ? -- insn' ) : >test-imm ( insn ? -- insn' )
(>compare-imm) [ vreg>integer ] dip next-vreg (>compare-imm) [ vreg>integer ] dip next-vreg
\ ##test-imm new-insn ; inline ##test-imm new-insn ; inline
: >test-imm-branch ( insn ? -- insn' ) : >test-imm-branch ( insn ? -- insn' )
(>compare-imm-branch) [ vreg>integer ] dip (>compare-imm-branch) [ vreg>integer ] dip
\ ##test-imm-branch new-insn ; inline ##test-imm-branch new-insn ; inline
M: ##test rewrite M: ##test rewrite
{ {

View File

@ -91,4 +91,4 @@ M: ##load-reference >expr obj>> <reference-expr> ;
M: ##phi >expr M: ##phi >expr
inputs>> values [ vreg>vn ] map inputs>> values [ vreg>vn ] map
basic-block get number>> prefix basic-block get number>> prefix
\ ##phi prefix ; ##phi prefix ;

View File

@ -24,7 +24,7 @@ M: ##shl-imm binary-constant-fold* drop shift ;
: binary-constant-fold ( insn -- insn' ) : binary-constant-fold ( insn -- insn' )
[ dst>> ] [ dst>> ]
[ [ src1>> vreg>integer ] [ src2>> ] [ ] tri binary-constant-fold* ] bi [ [ src1>> vreg>integer ] [ src2>> ] [ ] tri binary-constant-fold* ] bi
\ ##load-integer new-insn ; inline ##load-integer new-insn ; inline
: unary-constant-fold? ( insn -- ? ) : unary-constant-fold? ( insn -- ? )
src>> vreg>insn ##load-integer? ; inline src>> vreg>insn ##load-integer? ; inline
@ -36,4 +36,4 @@ M: ##neg unary-constant-fold* drop neg ;
: unary-constant-fold ( insn -- insn' ) : unary-constant-fold ( insn -- insn' )
[ dst>> ] [ [ src>> vreg>integer ] [ ] bi unary-constant-fold* ] bi [ dst>> ] [ [ src>> vreg>integer ] [ ] bi unary-constant-fold* ] bi
\ ##load-integer new-insn ; inline ##load-integer new-insn ; inline

View File

@ -20,8 +20,8 @@ IN: compiler.cfg.gvn.math
M: ##tagged>integer rewrite M: ##tagged>integer rewrite
[ dst>> ] [ src>> vreg>insn ] bi { [ dst>> ] [ src>> vreg>insn ] bi {
{ [ dup ##load-integer? ] [ val>> tag-fixnum \ ##load-integer new-insn ] } { [ dup ##load-integer? ] [ val>> tag-fixnum ##load-integer new-insn ] }
{ [ dup f-insn? ] [ drop \ f type-number \ ##load-integer new-insn ] } { [ dup f-insn? ] [ drop \ f type-number ##load-integer new-insn ] }
[ 2drop f ] [ 2drop f ]
} cond ; } cond ;
@ -81,14 +81,14 @@ M: ##add-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##add-imm? ] [ \ ##add-imm reassociate-arithmetic ] } { [ dup src1>> vreg>insn ##add-imm? ] [ ##add-imm reassociate-arithmetic ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
: sub-imm>add-imm ( insn -- insn' ) : sub-imm>add-imm ( insn -- insn' )
[ dst>> ] [ src1>> ] [ src2>> neg ] tri [ dst>> ] [ src1>> ] [ src2>> neg ] tri
dup immediate-arithmetic? dup immediate-arithmetic?
\ ##add-imm ?new-insn ; ##add-imm ?new-insn ;
M: ##sub-imm rewrite sub-imm>add-imm ; M: ##sub-imm rewrite sub-imm>add-imm ;
@ -97,14 +97,14 @@ M: ##sub-imm rewrite sub-imm>add-imm ;
src2>> -1 = ; src2>> -1 = ;
: mul-to-neg ( insn -- insn' ) : mul-to-neg ( insn -- insn' )
[ dst>> ] [ src1>> ] bi \ ##neg new-insn ; [ dst>> ] [ src1>> ] bi ##neg new-insn ;
! Convert ##mul-imm 2^X => ##shl-imm X ! Convert ##mul-imm 2^X => ##shl-imm X
: mul-to-shl? ( insn -- ? ) : mul-to-shl? ( insn -- ? )
src2>> power-of-2? ; src2>> power-of-2? ;
: mul-to-shl ( insn -- insn' ) : mul-to-shl ( insn -- insn' )
[ [ dst>> ] [ src1>> ] bi ] [ src2>> log2 ] bi \ ##shl-imm new-insn ; [ [ dst>> ] [ src1>> ] bi ] [ src2>> log2 ] bi ##shl-imm new-insn ;
! Distribution converts ! Distribution converts
! ##+-imm 2 1 X ! ##+-imm 2 1 X
@ -145,17 +145,17 @@ M: ##mul-imm rewrite
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup mul-to-neg? ] [ mul-to-neg ] } { [ dup mul-to-neg? ] [ mul-to-neg ] }
{ [ dup mul-to-shl? ] [ mul-to-shl ] } { [ dup mul-to-shl? ] [ mul-to-shl ] }
{ [ dup src1>> vreg>insn ##mul-imm? ] [ \ ##mul-imm reassociate-arithmetic ] } { [ dup src1>> vreg>insn ##mul-imm? ] [ ##mul-imm reassociate-arithmetic ] }
{ [ dup distribute-over-add? ] [ \ ##add-imm \ ##mul-imm distribute ] } { [ dup distribute-over-add? ] [ \ ##add-imm, \ ##mul-imm, distribute ] }
{ [ dup distribute-over-sub? ] [ \ ##sub-imm \ ##mul-imm distribute ] } { [ dup distribute-over-sub? ] [ \ ##sub-imm, \ ##mul-imm, distribute ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##and-imm rewrite M: ##and-imm rewrite
{ {
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##and-imm? ] [ \ ##and-imm reassociate-bitwise ] } { [ dup src1>> vreg>insn ##and-imm? ] [ ##and-imm reassociate-bitwise ] }
{ [ dup src2>> 0 = ] [ dst>> 0 \ ##load-integer new-insn ] } { [ dup src2>> 0 = ] [ dst>> 0 ##load-integer new-insn ] }
{ [ dup src2>> -1 = ] [ identity ] } { [ dup src2>> -1 = ] [ identity ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -163,18 +163,18 @@ M: ##and-imm rewrite
M: ##or-imm rewrite M: ##or-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup src2>> -1 = ] [ dst>> -1 \ ##load-integer new-insn ] } { [ dup src2>> -1 = ] [ dst>> -1 ##load-integer new-insn ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##or-imm? ] [ \ ##or-imm reassociate-bitwise ] } { [ dup src1>> vreg>insn ##or-imm? ] [ ##or-imm reassociate-bitwise ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##xor-imm rewrite M: ##xor-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup src2>> -1 = ] [ [ dst>> ] [ src1>> ] bi \ ##not new-insn ] } { [ dup src2>> -1 = ] [ [ dst>> ] [ src1>> ] bi ##not new-insn ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##xor-imm? ] [ \ ##xor-imm reassociate-bitwise ] } { [ dup src1>> vreg>insn ##xor-imm? ] [ ##xor-imm reassociate-bitwise ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -182,9 +182,9 @@ M: ##shl-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##shl-imm? ] [ \ ##shl-imm reassociate-shift ] } { [ dup src1>> vreg>insn ##shl-imm? ] [ ##shl-imm reassociate-shift ] }
{ [ dup distribute-over-add? ] [ \ ##add-imm \ ##shl-imm distribute ] } { [ dup distribute-over-add? ] [ \ ##add-imm, \ ##shl-imm, distribute ] }
{ [ dup distribute-over-sub? ] [ \ ##sub-imm \ ##shl-imm distribute ] } { [ dup distribute-over-sub? ] [ \ ##sub-imm, \ ##shl-imm, distribute ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -192,7 +192,7 @@ M: ##shr-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##shr-imm? ] [ \ ##shr-imm reassociate-shift ] } { [ dup src1>> vreg>insn ##shr-imm? ] [ ##shr-imm reassociate-shift ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -200,7 +200,7 @@ M: ##sar-imm rewrite
{ {
{ [ dup src2>> 0 = ] [ identity ] } { [ dup src2>> 0 = ] [ identity ] }
{ [ dup binary-constant-fold? ] [ binary-constant-fold ] } { [ dup binary-constant-fold? ] [ binary-constant-fold ] }
{ [ dup src1>> vreg>insn ##sar-imm? ] [ \ ##sar-imm reassociate-shift ] } { [ dup src1>> vreg>insn ##sar-imm? ] [ ##sar-imm reassociate-shift ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -217,8 +217,8 @@ M: ##sar-imm rewrite
M: ##add rewrite M: ##add rewrite
{ {
{ [ dup src2>> vreg-immediate-arithmetic? ] [ \ ##add-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-arithmetic? ] [ ##add-imm f insn>imm-insn ] }
{ [ dup src1>> vreg-immediate-arithmetic? ] [ \ ##add-imm t insn>imm-insn ] } { [ dup src1>> vreg-immediate-arithmetic? ] [ ##add-imm t insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -227,7 +227,7 @@ M: ##add rewrite
! ##sub 2 1 1 => ##load-integer 2 0 ! ##sub 2 1 1 => ##load-integer 2 0
: rewrite-subtraction-identity ( insn -- insn' ) : rewrite-subtraction-identity ( insn -- insn' )
dst>> 0 \ ##load-integer new-insn ; dst>> 0 ##load-integer new-insn ;
! ##load-integer 1 0 ! ##load-integer 1 0
! ##sub 3 1 2 ! ##sub 3 1 2
@ -237,61 +237,61 @@ M: ##add rewrite
src1>> vreg>insn zero-insn? ; src1>> vreg>insn zero-insn? ;
: sub-to-neg ( ##sub -- insn ) : sub-to-neg ( ##sub -- insn )
[ dst>> ] [ src2>> ] bi \ ##neg new-insn ; [ dst>> ] [ src2>> ] bi ##neg new-insn ;
M: ##sub rewrite M: ##sub rewrite
{ {
{ [ dup sub-to-neg? ] [ sub-to-neg ] } { [ dup sub-to-neg? ] [ sub-to-neg ] }
{ [ dup diagonal? ] [ rewrite-subtraction-identity ] } { [ dup diagonal? ] [ rewrite-subtraction-identity ] }
{ [ dup src2>> vreg-immediate-arithmetic? ] [ \ ##sub-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-arithmetic? ] [ ##sub-imm f insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##mul rewrite M: ##mul rewrite
{ {
{ [ dup src2>> vreg-immediate-arithmetic? ] [ \ ##mul-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-arithmetic? ] [ ##mul-imm f insn>imm-insn ] }
{ [ dup src1>> vreg-immediate-arithmetic? ] [ \ ##mul-imm t insn>imm-insn ] } { [ dup src1>> vreg-immediate-arithmetic? ] [ ##mul-imm t insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##and rewrite M: ##and rewrite
{ {
{ [ dup diagonal? ] [ identity ] } { [ dup diagonal? ] [ identity ] }
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##and-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##and-imm f insn>imm-insn ] }
{ [ dup src1>> vreg-immediate-bitwise? ] [ \ ##and-imm t insn>imm-insn ] } { [ dup src1>> vreg-immediate-bitwise? ] [ ##and-imm t insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##or rewrite M: ##or rewrite
{ {
{ [ dup diagonal? ] [ identity ] } { [ dup diagonal? ] [ identity ] }
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##or-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##or-imm f insn>imm-insn ] }
{ [ dup src1>> vreg-immediate-bitwise? ] [ \ ##or-imm t insn>imm-insn ] } { [ dup src1>> vreg-immediate-bitwise? ] [ ##or-imm t insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##xor rewrite M: ##xor rewrite
{ {
{ [ dup diagonal? ] [ dst>> 0 \ ##load-integer new-insn ] } { [ dup diagonal? ] [ dst>> 0 ##load-integer new-insn ] }
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##xor-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##xor-imm f insn>imm-insn ] }
{ [ dup src1>> vreg-immediate-bitwise? ] [ \ ##xor-imm t insn>imm-insn ] } { [ dup src1>> vreg-immediate-bitwise? ] [ ##xor-imm t insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##shl rewrite M: ##shl rewrite
{ {
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##shl-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##shl-imm f insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##shr rewrite M: ##shr rewrite
{ {
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##shr-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##shr-imm f insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
M: ##sar rewrite M: ##sar rewrite
{ {
{ [ dup src2>> vreg-immediate-bitwise? ] [ \ ##sar-imm f insn>imm-insn ] } { [ dup src2>> vreg-immediate-bitwise? ] [ ##sar-imm f insn>imm-insn ] }
[ drop f ] [ drop f ]
} cond ; } cond ;

View File

@ -11,5 +11,5 @@ M: ##replace rewrite
[ loc>> ] [ src>> vreg>insn ] bi [ loc>> ] [ src>> vreg>insn ] bi
dup literal-insn? [ dup literal-insn? [
insn>literal dup immediate-store? insn>literal dup immediate-store?
[ swap \ ##replace-imm new-insn ] [ 2drop f ] if [ swap ##replace-imm new-insn ] [ 2drop f ] if
] [ 2drop f ] if ; ] [ 2drop f ] if ;

View File

@ -3,7 +3,7 @@
USING: accessors combinators combinators.short-circuit arrays USING: accessors combinators combinators.short-circuit arrays
fry kernel layouts math namespaces sequences cpu.architecture fry kernel layouts math namespaces sequences cpu.architecture
math.bitwise math.order classes generalizations math.bitwise math.order classes generalizations
combinators.smart locals make alien.c-types io.binary grouping locals make alien.c-types io.binary grouping
math.vectors.simd.intrinsics math.vectors.simd.intrinsics
compiler.cfg compiler.cfg
compiler.cfg.registers compiler.cfg.registers
@ -27,7 +27,7 @@ IN: compiler.cfg.gvn.simd
[ [ dst>> ] [ src>> ] bi* ] [ [ dst>> ] [ src>> ] bi* ]
[ [ shuffle>> ] bi@ nths ] [ [ shuffle>> ] bi@ nths ]
[ drop rep>> ] [ drop rep>> ]
2tri \ ##shuffle-vector-imm new-insn 2tri ##shuffle-vector-imm new-insn
] [ 2drop f ] if ; ] [ 2drop f ] if ;
: (fold-shuffle-vector-imm) ( shuffle bytes -- bytes' ) : (fold-shuffle-vector-imm) ( shuffle bytes -- bytes' )
@ -35,7 +35,7 @@ IN: compiler.cfg.gvn.simd
: fold-shuffle-vector-imm ( outer inner -- insn' ) : fold-shuffle-vector-imm ( outer inner -- insn' )
[ [ dst>> ] [ shuffle>> ] bi ] [ obj>> ] bi* [ [ dst>> ] [ shuffle>> ] bi ] [ obj>> ] bi*
(fold-shuffle-vector-imm) \ ##load-reference new-insn ; (fold-shuffle-vector-imm) ##load-reference new-insn ;
M: ##shuffle-vector-imm rewrite M: ##shuffle-vector-imm rewrite
dup src>> vreg>insn { dup src>> vreg>insn {
@ -54,7 +54,7 @@ M: ##shuffle-vector-imm rewrite
: (fold-scalar>vector) ( insn bytes -- insn' ) : (fold-scalar>vector) ( insn bytes -- insn' )
[ [ dst>> ] [ rep>> rep-length ] bi ] dip <repetition> concat [ [ dst>> ] [ rep>> rep-length ] bi ] dip <repetition> concat
\ ##load-reference new-insn ; ##load-reference new-insn ;
: fold-scalar>vector ( outer inner -- insn' ) : fold-scalar>vector ( outer inner -- insn' )
over rep>> scalar-value (fold-scalar>vector) ; over rep>> scalar-value (fold-scalar>vector) ;
@ -69,7 +69,7 @@ M: ##scalar>vector rewrite
:: fold-gather-vector-2 ( insn src1 src2 -- insn ) :: fold-gather-vector-2 ( insn src1 src2 -- insn )
insn dst>> insn dst>>
src1 src2 [ insn rep>> scalar-value ] bi@ append src1 src2 [ insn rep>> scalar-value ] bi@ append
\ ##load-reference new-insn ; ##load-reference new-insn ;
: rewrite-gather-vector-2 ( insn -- insn/f ) : rewrite-gather-vector-2 ( insn -- insn/f )
dup [ src1>> vreg>insn ] [ src2>> vreg>insn ] bi { dup [ src1>> vreg>insn ] [ src2>> vreg>insn ] bi {
@ -85,9 +85,9 @@ M: ##gather-int-vector-2 rewrite rewrite-gather-vector-2 ;
insn dst>> insn dst>>
[ [
src1 src2 src3 src4 src1 src2 src3 src4
[ insn rep>> scalar-value ] 4 napply [ insn rep>> scalar-value % ] 4 napply
] B{ } append-outputs-as ] B{ } make
\ ##load-reference new-insn ; ##load-reference new-insn ;
: rewrite-gather-vector-4 ( insn -- insn/f ) : rewrite-gather-vector-4 ( insn -- insn/f )
dup { [ src1>> ] [ src2>> ] [ src3>> ] [ src4>> ] } cleave [ vreg>insn ] 4 napply dup { [ src1>> ] [ src2>> ] [ src3>> ] [ src4>> ] } cleave [ vreg>insn ] 4 napply
@ -102,7 +102,7 @@ M: ##gather-int-vector-4 rewrite rewrite-gather-vector-4 ;
: fold-shuffle-vector ( insn src1 src2 -- insn ) : fold-shuffle-vector ( insn src1 src2 -- insn )
[ dst>> ] [ obj>> ] [ obj>> ] tri* [ dst>> ] [ obj>> ] [ obj>> ] tri*
swap nths \ ##load-reference new-insn ; swap nths ##load-reference new-insn ;
M: ##shuffle-vector rewrite M: ##shuffle-vector rewrite
dup [ src>> vreg>insn ] [ shuffle>> vreg>insn ] bi dup [ src>> vreg>insn ] [ shuffle>> vreg>insn ] bi
@ -113,7 +113,7 @@ M: ##shuffle-vector rewrite
M: ##xor-vector rewrite M: ##xor-vector rewrite
dup diagonal? dup diagonal?
[ [ dst>> ] [ rep>> ] bi \ ##zero-vector new-insn ] [ drop f ] if ; [ [ dst>> ] [ rep>> ] bi ##zero-vector new-insn ] [ drop f ] if ;
: vector-not? ( insn -- ? ) : vector-not? ( insn -- ? )
{ {
@ -140,7 +140,7 @@ M: ##and-vector rewrite
[ src1>> vreg>insn vector-not-src ] [ src1>> vreg>insn vector-not-src ]
[ src2>> ] [ src2>> ]
[ rep>> ] [ rep>> ]
} cleave \ ##andn-vector new-insn } cleave ##andn-vector new-insn
] } ] }
{ [ dup src2>> vreg>insn vector-not? ] [ { [ dup src2>> vreg>insn vector-not? ] [
{ {
@ -148,7 +148,7 @@ M: ##and-vector rewrite
[ src2>> vreg>insn vector-not-src ] [ src2>> vreg>insn vector-not-src ]
[ src1>> ] [ src1>> ]
[ rep>> ] [ rep>> ]
} cleave \ ##andn-vector new-insn } cleave ##andn-vector new-insn
] } ] }
[ drop f ] [ drop f ]
} cond ; } cond ;
@ -160,5 +160,5 @@ M: ##andn-vector rewrite
[ src1>> vreg>insn vector-not-src ] [ src1>> vreg>insn vector-not-src ]
[ src2>> ] [ src2>> ]
[ rep>> ] [ rep>> ]
} cleave \ ##and-vector new-insn } cleave ##and-vector new-insn
] [ drop f ] if ; ] [ drop f ] if ;