Skip to content

Commit d14294a

Browse files
committed
Fix
1 parent 3015f1e commit d14294a

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

test/fft.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@ end
2222
Ns = (8,32,64,8)
2323

2424
# Complex tests
25-
for T in (ComplexF32, ComplexF64)
25+
for T in intersect(eltypes, [ComplexF32, ComplexF64])
2626
@testset "complex $T" begin
2727
# 1D out-of-place
2828
X = rand(T, Ns[1])
@@ -69,7 +69,7 @@ end
6969
end
7070

7171
# Real tests
72-
for T in (Float32, Float64)
72+
for T in (intersect(eltypes, [Float32, Float64]))
7373
@testset "real $T" begin
7474
X = rand(T, Ns[1])
7575
dX = gpu(X)
@@ -93,15 +93,15 @@ end
9393
end
9494

9595
# Wrapper convenience
96-
for T in (ComplexF32, ComplexF64)
96+
for T in intersect(eltypes, [ComplexF32, ComplexF64])
9797
X = gpu(rand(T, Ns[1], Ns[2]))
9898
Y = fft(X)
9999
cmp(Y, fft(Array(X)))
100100
Z = ifft(Y)
101101
cmp(Z, Array(X))
102102
end
103103

104-
for T in (Float32, Float64)
104+
for T in intersect(eltypes, [Float32, Float64])
105105
X = gpu(rand(T, Ns[1], Ns[2]))
106106
Y = rfft(X)
107107
cmp(Y, rfft(Array(X), (1,))) # Compare with 1D FFT along first dim, not multi-dimensional FFT

0 commit comments

Comments
 (0)