141
141
@inline _mul_partials (a:: Partials{0,A} , b:: Partials{N,B} , afactor, bfactor) where {N,A,B} = bfactor * b
142
142
@inline _mul_partials (a:: Partials{N,A} , b:: Partials{0,B} , afactor, bfactor) where {N,A,B} = afactor * a
143
143
144
- const SIMDFloat = Union{Float64, Float32}
145
- const SIMDInt = Union{
146
- Int128, Int64, Int32, Int16, Int8,
147
- UInt128, UInt64, UInt32, UInt16, UInt8,
148
- }
149
- const SIMDType = Union{SIMDFloat, SIMDInt}
150
-
151
144
# #################################
152
145
# Generated Functions on NTuples #
153
146
# #################################
171
164
@inline rand_tuple (:: AbstractRNG , :: Type{Tuple{}} ) = tuple ()
172
165
@inline rand_tuple (:: Type{Tuple{}} ) = tuple ()
173
166
174
- iszero_tuple (tup:: NTuple{N,V} ) where {N, V<: SIMDType } = sum (Vec (tup) != zero (V)) == 0
175
167
@generated function iszero_tuple (tup:: NTuple{N,V} ) where {N,V}
176
168
ex = Expr (:&& , [:(z == tup[$ i]) for i= 1 : N]. .. )
177
169
return quote
@@ -213,14 +205,15 @@ const SIMDInt = Union{
213
205
}
214
206
const SIMDType = Union{SIMDFloat, SIMDInt}
215
207
const NT{N,T} = NTuple{N,T}
208
+ using SIMD
216
209
217
210
# SIMD implementation
218
- @inline add_tuples (a:: NT{N,T} , b:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (Vec (a) + Vec (b))
219
- @inline sub_tuples (a:: NT{N,T} , b:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (Vec (a) - Vec (b))
220
- @inline scale_tuple (tup:: NT{N,T} , x:: T ) where {N, T<: SIMDType } = Tuple (Vec (tup) * x)
221
- @inline div_tuple_by_scalar (tup:: NT{N,T} , x:: T ) where {N, T<: SIMDFloat } = Tuple (Vec (tup) / x)
222
- @inline minus_tuple (tup:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (- Vec (tup))
223
- @inline mul_tuples (a:: NT{N,T} , b:: NT{N,T} , af:: T , bf:: T ) where {N, T<: SIMDType } = Tuple (muladd (af , Vec (a), bf * Vec (b)))
211
+ add_tuples (a:: NT{N,T} , b:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (Vec (a) + Vec (b))
212
+ sub_tuples (a:: NT{N,T} , b:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (Vec (a) - Vec (b))
213
+ scale_tuple (tup:: NT{N,T} , x:: T ) where {N, T<: SIMDType } = Tuple (Vec (tup) * x)
214
+ div_tuple_by_scalar (tup:: NT{N,T} , x:: T ) where {N, T<: SIMDFloat } = Tuple (Vec (tup) / x)
215
+ minus_tuple (tup:: NT{N,T} ) where {N, T<: SIMDType } = Tuple (- Vec (tup))
216
+ mul_tuples (a:: NT{N,T} , b:: NT{N,T} , af:: T , bf:: T ) where {N, T<: SIMDType } = Tuple (muladd (Vec {N,T} (af) , Vec (a), Vec {N,T} (bf) * Vec (b)))
224
217
225
218
226
219
# Fallback implementations
@@ -229,7 +222,7 @@ const NT{N,T} = NTuple{N,T}
229
222
@generated scale_tuple (tup:: NT{N} , x) where N = tupexpr (i -> :(tup[$ i] * x), N)
230
223
@generated div_tuple_by_scalar (tup:: NT{N} , x) where N = tupexpr (i -> :(tup[$ i] / x), N)
231
224
@generated minus_tuple (tup:: NT{N} ) where N = tupexpr (i -> :(- tup[$ i]), N)
232
- @generated mul_tuples (a:: NT{N} , b:: NT{N} , af, bf) where N = tupexpr (i -> :(muladd (af, a[$ i], bf * b[$ i])), N)
225
+ @generated mul_tuples (a:: NT{N} , b:: NT{N} , af, bf) where N = tupexpr (i -> :((af * a[$ i]) + ( bf * b[$ i])), N)
233
226
234
227
# ##################
235
228
# Pretty Printing #
0 commit comments