Skip to content

Commit 2051f0c

Browse files
committed
use cleaner extractelement
1 parent 7d6ac8d commit 2051f0c

File tree

2 files changed

+6
-9
lines changed

2 files changed

+6
-9
lines changed

Project.toml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@ SIMDMath = "5443be0b-e40a-4f70-a07e-dcd652efc383"
77

88
[compat]
99
julia = "1.8"
10-
SIMDMath = "0.2.1"
10+
SIMDMath = "0.2.3"
1111

1212
[extras]
1313
Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"

src/Airy/cairy.jl

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -259,17 +259,17 @@ function airyai_power_series(z::Complex{T}) where T
259259
z2 = z * z
260260
z3 = z2 * z
261261
p = SIMDMath.horner_simd(z3, pack_AIRYAI_POW_COEF)
262-
ai = muladd(-complex(p.re[2].value, p.im[2].value), z, complex(p.re[1].value, p.im[1].value))
263-
aip = muladd(complex(p.re[4].value, p.im[4].value), z2, -complex(p.re[3].value, p.im[3].value))
262+
ai = muladd(-p[2], z, p[1])
263+
aip = muladd(p[4], z2, -p[3])
264264
return ai, aip
265265
end
266266

267267
function airybi_power_series(z::Complex{T}) where T
268268
z2 = z * z
269269
z3 = z2 * z
270270
p = SIMDMath.horner_simd(z3, pack_AIRYBI_POW_COEF)
271-
bi = muladd(complex(p.re[2].value, p.im[2].value), z, complex(p.re[1].value, p.im[1].value))
272-
bip = muladd(complex(p.re[4].value, p.im[4].value), z2, complex(p.re[3].value, p.im[3].value))
271+
bi = muladd(p[2], z, p[1])
272+
bip = muladd(p[4], z2, p[3])
273273
return bi, bip
274274
end
275275

@@ -368,10 +368,7 @@ end
368368
a = SIMDMath.fadd(pvec1, zvec)
369369
b = SIMDMath.fsub(pvec1, zvec)
370370

371-
A = complex(b.re[1].value, b.im[1].value)
372-
B = complex(a.re[1].value, a.im[1].value)
373-
C = complex(b.re[2].value, b.im[2].value)
374-
D = complex(a.re[2].value, a.im[2].value)
371+
A, B, C, D = b[1], a[1], b[2], a[2]
375372
return A, B, C, D
376373
end
377374

0 commit comments

Comments
 (0)