1 ! Copyright (C) 2009 Slava Pestov.
2 ! See http://factorcode.org/license.txt for BSD license.
3 USING: accessors assocs byte-arrays combinators compiler.cfg.builder
4 continuations fry sequences compiler.tree.propagation.info
5 cpu.architecture kernel words make math math.intervals
6 math.vectors.simd.intrinsics namespaces ;
7 IN: compiler.tree.propagation.simd
9 CONSTANT: vector>vector-intrinsics
41 (simd-vshuffle-elements)
63 CONSTANT: vector-other-intrinsics
75 : vector-intrinsics ( -- x )
76 vector>vector-intrinsics vector-other-intrinsics append ;
78 vector>vector-intrinsics [ { byte-array } "default-output-classes" set-word-prop ] each
80 : scalar-output-class ( rep -- class )
82 literal>> scalar-rep-of {
83 { float-rep [ float ] }
84 { double-rep [ float ] }
90 \ (simd-sum) [ nip scalar-output-class ] "outputs" set-word-prop
92 \ (simd-v.) [ 2nip scalar-output-class ] "outputs" set-word-prop
98 } [ { boolean } "default-output-classes" set-word-prop ] each
100 \ (simd-select) [ 2nip scalar-output-class ] "outputs" set-word-prop
103 real [0,inf] <class/interval-info> value-info-intersect
104 ] "outputs" set-word-prop
106 : clone-with-value-infos ( node -- node' )
107 clone dup in-d>> [ dup value-info ] H{ } map>assoc >>info ;
109 : try-intrinsic ( node intrinsic-quot -- ? )
111 _ clone-with-value-infos
112 _ with-dummy-cfg-builder
114 ] [ drop f ] recover ;
116 : inline-unless-intrinsic ( word -- )
118 _ swap over "intrinsic" word-prop
119 "always-inline-simd-intrinsics" get not swap and
120 ! word node intrinsic
121 [ try-intrinsic [ drop f ] [ def>> ] if ]
124 "custom-inlining" set-word-prop ;
126 vector-intrinsics [ inline-unless-intrinsic ] each