Skip to content

Commit 7e8f41c

Browse files
authored
test: WeightInitializers test mark broken (#1278)
1 parent 01635ec commit 7e8f41c

File tree

1 file changed

+2
-5
lines changed

1 file changed

+2
-5
lines changed

lib/WeightInitializers/test/initializers_tests.jl

Lines changed: 2 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -359,12 +359,9 @@ end
359359
v = kaiming_normal(rng, n_in, n_out)
360360
σ2 = sqrt(2 / n_out)
361361

362-
if (backend == "cuda" || backend == "amdgpu") && rng isa GPUArrays.RNG
363-
@test_broken 0.9σ2 < std(v) < 1.1σ2
364-
else
365-
@test 0.9σ2 < std(v) < 1.1σ2
366-
end
362+
@test 0.9σ2 < std(v) < 1.1σ2 broken = rng isa GPUArrays.RNG
367363
end
364+
368365
# Type
369366
@test eltype(kaiming_uniform(rng, 3, 4; gain=1.5f0)) == Float32
370367
@test eltype(kaiming_normal(rng, 3, 4; gain=1.5f0)) == Float32

0 commit comments

Comments
 (0)