Skip to content

Commit ed7129e

Browse files
committed
Only broken on GPU
1 parent 7f818e9 commit ed7129e

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/testsuite/random.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
cpu_rng = Random.default_rng()
88

99
@testset "rand" begin # uniform
10-
for T in eltypes, d in (10, (10, 10), (128, 128))
10+
@testset "$d $T" for T in eltypes, d in (10, (10, 10), (128, 128))
1111
A = AT{T}(undef, d)
1212
B = copy(A)
1313
rand!(rng, A)
@@ -22,7 +22,7 @@
2222
rand!(rng, A)
2323
Random.seed!(rng, 1)
2424
rand!(rng, B)
25-
@test Array(A) == Array(B) broken=prod(d) > length(rng.state)
25+
@test Array(A) == Array(B) broken=(rng != cpu_rng) && (prod(d) > length(rng.state))
2626

2727
if rng != cpu_rng
2828
rand!(cpu_rng, A)
@@ -47,7 +47,7 @@
4747
@testset "randn" begin # normally-distributed
4848
# XXX: randn calls sqrt, and Base's sqrt(::Complex) performs
4949
# checked type conversions that throw boxed numbers.
50-
for T in filter(isrealfloattype, eltypes), d in (2, (2, 2), (128, 128))
50+
@testset "$d $T" for T in filter(isrealfloattype, eltypes), d in (2, (2, 2), (128, 128))
5151
A = AT{T}(undef, d)
5252
B = copy(A)
5353
randn!(rng, A)
@@ -62,7 +62,7 @@
6262
randn!(rng, A)
6363
Random.seed!(rng, 1)
6464
randn!(rng, B)
65-
@test Array(A) == Array(B) broken=prod(d) > (2 * length(rng.state))
65+
@test Array(A) == Array(B) broken=(rng != cpu_rng) && (prod(d) > (2 * length(rng.state)))
6666

6767
if rng != cpu_rng
6868
randn!(cpu_rng, A)

0 commit comments

Comments
 (0)