;; Copyright (c) 2010-2015, Intel Corporation ;; All rights reserved. ;; ;; Redistribution and use in source and binary forms, with or without ;; modification, are permitted provided that the following conditions are ;; met: ;; ;; * Redistributions of source code must retain the above copyright ;; notice, this list of conditions and the following disclaimer. ;; ;; * Redistributions in binary form must reproduce the above copyright ;; notice, this list of conditions and the following disclaimer in the ;; documentation and/or other materials provided with the distribution. ;; ;; * Neither the name of Intel Corporation nor the names of its ;; contributors may be used to endorse or promote products derived from ;; this software without specific prior written permission. ;; ;; ;; THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS ;; IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED ;; TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A ;; PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER ;; OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, ;; EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, ;; PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR ;; PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF ;; LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING ;; NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS ;; SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ; Define common 4-wide stuff define(`WIDTH',`4') define(`MASK',`i32') include(`util.m4') stdlib_core() packed_load_and_store() scans() int64minmax() saturation_arithmetic() include(`target-sse4-common.ll') ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; half conversion routines declare float @__half_to_float_uniform(i16 %v) nounwind readnone declare @__half_to_float_varying( %v) nounwind readnone declare i16 @__float_to_half_uniform(float %v) nounwind readnone declare @__float_to_half_varying( %v) nounwind readnone ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; rcp declare <4 x float> @llvm.x86.sse.rcp.ps(<4 x float>) nounwind readnone define <4 x float> @__rcp_varying_float(<4 x float>) nounwind readonly alwaysinline { ; do one N-R iteration to improve precision ; float iv = __rcp_v(v); ; return iv * (2. - v * iv); %call = call <4 x float> @llvm.x86.sse.rcp.ps(<4 x float> %0) %v_iv = fmul <4 x float> %0, %call %two_minus = fsub <4 x float> , %v_iv %iv_mul = fmul <4 x float> %call, %two_minus ret <4 x float> %iv_mul } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ; rsqrt declare <4 x float> @llvm.x86.sse.rsqrt.ps(<4 x float>) nounwind readnone define <4 x float> @__rsqrt_varying_float(<4 x float> %v) nounwind readonly alwaysinline { ; float is = __rsqrt_v(v); %is = call <4 x float> @llvm.x86.sse.rsqrt.ps(<4 x float> %v) ; Newton-Raphson iteration to improve precision ; return 0.5 * is * (3. - (v * is) * is); %v_is = fmul <4 x float> %v, %is %v_is_is = fmul <4 x float> %v_is, %is %three_sub = fsub <4 x float> , %v_is_is %is_mul = fmul <4 x float> %is, %three_sub %half_scale = fmul <4 x float> , %is_mul ret <4 x float> %half_scale } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; sqrt declare <4 x float> @llvm.x86.sse.sqrt.ps(<4 x float>) nounwind readnone define <4 x float> @__sqrt_varying_float(<4 x float>) nounwind readonly alwaysinline { %call = call <4 x float> @llvm.x86.sse.sqrt.ps(<4 x float> %0) ret <4 x float> %call } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; double precision sqrt declare <2 x double> @llvm.x86.sse2.sqrt.pd(<2 x double>) nounwind readnone define <4 x double> @__sqrt_varying_double(<4 x double>) nounwind alwaysinline { unary2to4(ret, double, @llvm.x86.sse2.sqrt.pd, %0) ret <4 x double> %ret } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; rounding floats declare <4 x float> @llvm.x86.sse41.round.ps(<4 x float>, i32) nounwind readnone define <4 x float> @__round_varying_float(<4 x float>) nounwind readonly alwaysinline { ; roundps, round mode nearest 0b00 | don't signal precision exceptions 0b1000 = 8 %call = call <4 x float> @llvm.x86.sse41.round.ps(<4 x float> %0, i32 8) ret <4 x float> %call } define <4 x float> @__floor_varying_float(<4 x float>) nounwind readonly alwaysinline { ; roundps, round down 0b01 | don't signal precision exceptions 0b1001 = 9 %call = call <4 x float> @llvm.x86.sse41.round.ps(<4 x float> %0, i32 9) ret <4 x float> %call } define <4 x float> @__ceil_varying_float(<4 x float>) nounwind readonly alwaysinline { ; roundps, round up 0b10 | don't signal precision exceptions 0b1010 = 10 %call = call <4 x float> @llvm.x86.sse41.round.ps(<4 x float> %0, i32 10) ret <4 x float> %call } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; rounding doubles declare <2 x double> @llvm.x86.sse41.round.pd(<2 x double>, i32) nounwind readnone define <4 x double> @__round_varying_double(<4 x double>) nounwind readonly alwaysinline { round2to4double(%0, 8) } define <4 x double> @__floor_varying_double(<4 x double>) nounwind readonly alwaysinline { ; roundpd, round down 0b01 | don't signal precision exceptions 0b1001 = 9 round2to4double(%0, 9) } define <4 x double> @__ceil_varying_double(<4 x double>) nounwind readonly alwaysinline { ; roundpd, round up 0b10 | don't signal precision exceptions 0b1010 = 10 round2to4double(%0, 10) } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; float min/max declare <4 x float> @llvm.x86.sse.max.ps(<4 x float>, <4 x float>) nounwind readnone declare <4 x float> @llvm.x86.sse.min.ps(<4 x float>, <4 x float>) nounwind readnone define <4 x float> @__max_varying_float(<4 x float>, <4 x float>) nounwind readonly alwaysinline { %call = call <4 x float> @llvm.x86.sse.max.ps(<4 x float> %0, <4 x float> %1) ret <4 x float> %call } define <4 x float> @__min_varying_float(<4 x float>, <4 x float>) nounwind readonly alwaysinline { %call = call <4 x float> @llvm.x86.sse.min.ps(<4 x float> %0, <4 x float> %1) ret <4 x float> %call } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; double precision min/max declare <2 x double> @llvm.x86.sse2.max.pd(<2 x double>, <2 x double>) nounwind readnone declare <2 x double> @llvm.x86.sse2.min.pd(<2 x double>, <2 x double>) nounwind readnone define <4 x double> @__min_varying_double(<4 x double>, <4 x double>) nounwind readnone { binary2to4(ret, double, @llvm.x86.sse2.min.pd, %0, %1) ret <4 x double> %ret } define <4 x double> @__max_varying_double(<4 x double>, <4 x double>) nounwind readnone { binary2to4(ret, double, @llvm.x86.sse2.max.pd, %0, %1) ret <4 x double> %ret } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; int32 min/max define <4 x i32> @__min_varying_int32(<4 x i32>, <4 x i32>) nounwind readonly alwaysinline { %call = call <4 x i32> @llvm.x86.sse41.pminsd(<4 x i32> %0, <4 x i32> %1) ret <4 x i32> %call } define <4 x i32> @__max_varying_int32(<4 x i32>, <4 x i32>) nounwind readonly alwaysinline { %call = call <4 x i32> @llvm.x86.sse41.pmaxsd(<4 x i32> %0, <4 x i32> %1) ret <4 x i32> %call } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ; unsigned int min/max define <4 x i32> @__min_varying_uint32(<4 x i32>, <4 x i32>) nounwind readonly alwaysinline { %call = call <4 x i32> @llvm.x86.sse41.pminud(<4 x i32> %0, <4 x i32> %1) ret <4 x i32> %call } define <4 x i32> @__max_varying_uint32(<4 x i32>, <4 x i32>) nounwind readonly alwaysinline { %call = call <4 x i32> @llvm.x86.sse41.pmaxud(<4 x i32> %0, <4 x i32> %1) ret <4 x i32> %call } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; svml stuff include(`svml.m4') ;; single precision svml_declare(float,f4,4) svml_define(float,f4,4,f) ;; double precision svml_declare(double,2,2) svml_define_x(double,2,2,d,4) ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; mask handling declare i32 @llvm.x86.sse.movmsk.ps(<4 x float>) nounwind readnone define i64 @__movmsk(<4 x i32>) nounwind readnone alwaysinline { %floatmask = bitcast <4 x i32> %0 to <4 x float> %v = call i32 @llvm.x86.sse.movmsk.ps(<4 x float> %floatmask) nounwind readnone %v64 = zext i32 %v to i64 ret i64 %v64 } define i1 @__any(<4 x i32>) nounwind readnone alwaysinline { %floatmask = bitcast <4 x i32> %0 to <4 x float> %v = call i32 @llvm.x86.sse.movmsk.ps(<4 x float> %floatmask) nounwind readnone %cmp = icmp ne i32 %v, 0 ret i1 %cmp } define i1 @__all(<4 x i32>) nounwind readnone alwaysinline { %floatmask = bitcast <4 x i32> %0 to <4 x float> %v = call i32 @llvm.x86.sse.movmsk.ps(<4 x float> %floatmask) nounwind readnone %cmp = icmp eq i32 %v, 15 ret i1 %cmp } define i1 @__none(<4 x i32>) nounwind readnone alwaysinline { %floatmask = bitcast <4 x i32> %0 to <4 x float> %v = call i32 @llvm.x86.sse.movmsk.ps(<4 x float> %floatmask) nounwind readnone %cmp = icmp eq i32 %v, 0 ret i1 %cmp } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal ops / reductions ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal float ops declare <4 x float> @llvm.x86.sse3.hadd.ps(<4 x float>, <4 x float>) nounwind readnone define float @__reduce_add_float(<4 x float>) nounwind readonly alwaysinline { %v1 = call <4 x float> @llvm.x86.sse3.hadd.ps(<4 x float> %0, <4 x float> %0) %v2 = call <4 x float> @llvm.x86.sse3.hadd.ps(<4 x float> %v1, <4 x float> %v1) %scalar = extractelement <4 x float> %v2, i32 0 ret float %scalar } define float @__reduce_min_float(<4 x float>) nounwind readnone alwaysinline { reduce4(float, @__min_varying_float, @__min_uniform_float) } define float @__reduce_max_float(<4 x float>) nounwind readnone alwaysinline { reduce4(float, @__max_varying_float, @__max_uniform_float) } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal double ops define double @__reduce_add_double(<4 x double>) nounwind readnone alwaysinline { %v0 = shufflevector <4 x double> %0, <4 x double> undef, <2 x i32> %v1 = shufflevector <4 x double> %0, <4 x double> undef, <2 x i32> %sum = fadd <2 x double> %v0, %v1 %e0 = extractelement <2 x double> %sum, i32 0 %e1 = extractelement <2 x double> %sum, i32 1 %m = fadd double %e0, %e1 ret double %m } define double @__reduce_min_double(<4 x double>) nounwind readnone alwaysinline { reduce4(double, @__min_varying_double, @__min_uniform_double) } define double @__reduce_max_double(<4 x double>) nounwind readnone alwaysinline { reduce4(double, @__max_varying_double, @__max_uniform_double) } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal int8 ops declare <2 x i64> @llvm.x86.sse2.psad.bw(<16 x i8>, <16 x i8>) nounwind readnone define i16 @__reduce_add_int8(<4 x i8>) nounwind readnone alwaysinline { %wide8 = shufflevector <4 x i8> %0, <4 x i8> zeroinitializer, <16 x i32> %rv = call <2 x i64> @llvm.x86.sse2.psad.bw(<16 x i8> %wide8, <16 x i8> zeroinitializer) %r0 = extractelement <2 x i64> %rv, i32 0 %r1 = extractelement <2 x i64> %rv, i32 1 %r = add i64 %r0, %r1 %r16 = trunc i64 %r to i16 ret i16 %r16 } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal int16 ops define internal <4 x i16> @__add_varying_i16(<4 x i16>, <4 x i16>) nounwind readnone alwaysinline { %r = add <4 x i16> %0, %1 ret <4 x i16> %r } define internal i16 @__add_uniform_i16(i16, i16) nounwind readnone alwaysinline { %r = add i16 %0, %1 ret i16 %r } define i16 @__reduce_add_int16(<4 x i16>) nounwind readnone alwaysinline { reduce4(i16, @__add_varying_i16, @__add_uniform_i16) } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal int32 ops ;; reduction functions define i32 @__reduce_add_int32(<4 x i32> %v) nounwind readnone alwaysinline { %v1 = shufflevector <4 x i32> %v, <4 x i32> undef, <4 x i32> %m1 = add <4 x i32> %v1, %v %m1a = extractelement <4 x i32> %m1, i32 0 %m1b = extractelement <4 x i32> %m1, i32 1 %sum = add i32 %m1a, %m1b ret i32 %sum } define i32 @__reduce_min_int32(<4 x i32>) nounwind readnone alwaysinline { reduce4(i32, @__min_varying_int32, @__min_uniform_int32) } define i32 @__reduce_max_int32(<4 x i32>) nounwind readnone alwaysinline { reduce4(i32, @__max_varying_int32, @__max_uniform_int32) } define i32 @__reduce_min_uint32(<4 x i32>) nounwind readnone alwaysinline { reduce4(i32, @__min_varying_uint32, @__min_uniform_uint32) } define i32 @__reduce_max_uint32(<4 x i32>) nounwind readnone alwaysinline { reduce4(i32, @__max_varying_uint32, @__max_uniform_uint32) } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; horizontal int64 ops ;; reduction functions define i64 @__reduce_add_int64(<4 x i64>) nounwind readnone alwaysinline { %v0 = shufflevector <4 x i64> %0, <4 x i64> undef, <2 x i32> %v1 = shufflevector <4 x i64> %0, <4 x i64> undef, <2 x i32> %sum = add <2 x i64> %v0, %v1 %e0 = extractelement <2 x i64> %sum, i32 0 %e1 = extractelement <2 x i64> %sum, i32 1 %m = add i64 %e0, %e1 ret i64 %m } define i64 @__reduce_min_int64(<4 x i64>) nounwind readnone alwaysinline { reduce4(i64, @__min_varying_int64, @__min_uniform_int64) } define i64 @__reduce_max_int64(<4 x i64>) nounwind readnone alwaysinline { reduce4(i64, @__max_varying_int64, @__max_uniform_int64) } define i64 @__reduce_min_uint64(<4 x i64>) nounwind readnone alwaysinline { reduce4(i64, @__min_varying_uint64, @__min_uniform_uint64) } define i64 @__reduce_max_uint64(<4 x i64>) nounwind readnone alwaysinline { reduce4(i64, @__max_varying_uint64, @__max_uniform_uint64) } reduce_equal(4) ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; unaligned loads/loads+broadcasts masked_load(i8, 1) masked_load(i16, 2) masked_load(i32, 4) masked_load(float, 4) masked_load(i64, 8) masked_load(double, 8) ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; masked store gen_masked_store(i8) gen_masked_store(i16) gen_masked_store(i32) gen_masked_store(i64) masked_store_float_double() ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; masked store blend masked_store_blend_8_16_by_4() declare <4 x float> @llvm.x86.sse41.blendvps(<4 x float>, <4 x float>, <4 x float>) nounwind readnone define void @__masked_store_blend_i32(<4 x i32>* nocapture, <4 x i32>, <4 x i32> %mask) nounwind alwaysinline { %mask_as_float = bitcast <4 x i32> %mask to <4 x float> %oldValue = load PTR_OP_ARGS(`<4 x i32>') %0, align 4 %oldAsFloat = bitcast <4 x i32> %oldValue to <4 x float> %newAsFloat = bitcast <4 x i32> %1 to <4 x float> %blend = call <4 x float> @llvm.x86.sse41.blendvps(<4 x float> %oldAsFloat, <4 x float> %newAsFloat, <4 x float> %mask_as_float) %blendAsInt = bitcast <4 x float> %blend to <4 x i32> store <4 x i32> %blendAsInt, <4 x i32>* %0, align 4 ret void } define void @__masked_store_blend_i64(<4 x i64>* nocapture %ptr, <4 x i64> %new, <4 x i32> %i32mask) nounwind alwaysinline { %oldValue = load PTR_OP_ARGS(`<4 x i64>') %ptr, align 8 %mask = bitcast <4 x i32> %i32mask to <4 x float> ; Do 4x64-bit blends by doing two <4 x i32> blends, where the <4 x i32> values ; are actually bitcast <2 x i64> values ; ; set up the first two 64-bit values %old01 = shufflevector <4 x i64> %oldValue, <4 x i64> undef, <2 x i32> %old01f = bitcast <2 x i64> %old01 to <4 x float> %new01 = shufflevector <4 x i64> %new, <4 x i64> undef, <2 x i32> %new01f = bitcast <2 x i64> %new01 to <4 x float> ; compute mask--note that the indices 0 and 1 are doubled-up %mask01 = shufflevector <4 x float> %mask, <4 x float> undef, <4 x i32> ; and blend the two of the values %result01f = call <4 x float> @llvm.x86.sse41.blendvps(<4 x float> %old01f, <4 x float> %new01f, <4 x float> %mask01) %result01 = bitcast <4 x float> %result01f to <2 x i64> ; and again %old23 = shufflevector <4 x i64> %oldValue, <4 x i64> undef, <2 x i32> %old23f = bitcast <2 x i64> %old23 to <4 x float> %new23 = shufflevector <4 x i64> %new, <4 x i64> undef, <2 x i32> %new23f = bitcast <2 x i64> %new23 to <4 x float> ; compute mask--note that the values 2 and 3 are doubled-up %mask23 = shufflevector <4 x float> %mask, <4 x float> undef, <4 x i32> ; and blend the two of the values %result23f = call <4 x float> @llvm.x86.sse41.blendvps(<4 x float> %old23f, <4 x float> %new23f, <4 x float> %mask23) %result23 = bitcast <4 x float> %result23f to <2 x i64> ; reconstruct the final <4 x i64> vector %final = shufflevector <2 x i64> %result01, <2 x i64> %result23, <4 x i32> store <4 x i64> %final, <4 x i64> * %ptr, align 8 ret void } ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; gather/scatter ; define these with the macros from stdlib.m4 gen_gather_factored(i8) gen_gather_factored(i16) gen_gather_factored(i32) gen_gather_factored(float) gen_gather_factored(i64) gen_gather_factored(double) gen_scatter(i8) gen_scatter(i16) gen_scatter(i32) gen_scatter(float) gen_scatter(i64) gen_scatter(double) ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; int8/int16 builtins define_avgs() ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;; ;; reciprocals in double precision, if supported rsqrtd_decl() rcpd_decl() transcendetals_decl() trigonometry_decl()