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 compiler.tree.propagation.nodes cpu.architecture kernel layouts words
6 make math math.intervals math.vectors.simd.intrinsics namespaces ;
7 IN: compiler.tree.propagation.simd
9 CONSTANT: vector>vector-intrinsics
41 (simd-vshuffle-elements)
42 (simd-vshuffle2-elements)
64 CONSTANT: vector-other-intrinsics
77 : vector-intrinsics ( -- x )
78 vector>vector-intrinsics vector-other-intrinsics append ;
80 vector>vector-intrinsics [ { byte-array } "default-output-classes" set-word-prop ] each
82 : scalar-output-class ( rep -- class )
84 literal>> scalar-rep-of {
85 { float-rep [ float ] }
86 { double-rep [ float ] }
87 { longlong-scalar-rep [ integer ] }
88 { ulonglong-scalar-rep [ integer ] }
89 { int-scalar-rep [ cell 8 = [ fixnum ] [ integer ] if ] }
90 { uint-scalar-rep [ cell 8 = [ fixnum ] [ integer ] if ] }
96 \ (simd-sum) [ nip scalar-output-class ] "outputs" set-word-prop
98 \ (simd-v.) [ 2nip scalar-output-class ] "outputs" set-word-prop
104 } [ { boolean } "default-output-classes" set-word-prop ] each
106 \ (simd-select) [ 2nip scalar-output-class ] "outputs" set-word-prop
109 real [0,inf] <class/interval-info> value-info-intersect
110 ] "outputs" set-word-prop
112 \ (simd-vgetmask) { fixnum } "default-output-classes" set-word-prop
114 : clone-with-value-infos ( node -- node' )
115 clone dup in-d>> extract-value-info >>info ;
117 : try-intrinsic ( node intrinsic-quot -- ? )
119 _ clone-with-value-infos
120 _ with-dummy-cfg-builder
122 ] [ drop f ] recover ;
124 : inline-unless-intrinsic ( word -- )
126 _ swap over "intrinsic" word-prop
127 "always-inline-simd-intrinsics" get not swap and
128 ! word node intrinsic
129 [ try-intrinsic [ drop f ] [ def>> ] if ]
132 "custom-inlining" set-word-prop ;
134 vector-intrinsics [ inline-unless-intrinsic ] each