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
38 (simd-vshuffle-elements)
60 CONSTANT: vector-other-intrinsics
71 : vector-intrinsics ( -- x )
72 vector>vector-intrinsics vector-other-intrinsics append ;
74 vector>vector-intrinsics [ { byte-array } "default-output-classes" set-word-prop ] each
76 : scalar-output-class ( rep -- class )
78 literal>> scalar-rep-of {
79 { float-rep [ float ] }
80 { double-rep [ float ] }
86 \ (simd-sum) [ nip scalar-output-class ] "outputs" set-word-prop
88 \ (simd-v.) [ 2nip scalar-output-class ] "outputs" set-word-prop
94 } [ { boolean } "default-output-classes" set-word-prop ] each
96 \ (simd-select) [ 2nip scalar-output-class ] "outputs" set-word-prop
99 real [0,inf] <class/interval-info> value-info-intersect
100 ] "outputs" set-word-prop
102 : clone-with-value-infos ( node -- node' )
103 clone dup in-d>> [ dup value-info ] H{ } map>assoc >>info ;
105 : try-intrinsic ( node intrinsic-quot -- ? )
107 _ clone-with-value-infos
108 _ with-dummy-cfg-builder
110 ] [ drop f ] recover ;
112 : inline-unless-intrinsic ( word -- )
114 _ swap over "intrinsic" word-prop
115 "always-inline-simd-intrinsics" get not swap and
116 ! word node intrinsic
117 [ try-intrinsic [ drop f ] [ def>> ] if ]
120 "custom-inlining" set-word-prop ;
122 vector-intrinsics [ inline-unless-intrinsic ] each