Skip to content

Commit e45441e

Browse files
committed
More cleanup
1 parent f0c5d57 commit e45441e

File tree

5 files changed

+221
-275
lines changed

5 files changed

+221
-275
lines changed

test/chainrules.jl

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
using MatrixAlgebraKit
22
using Test
3-
using StableRNGs
43

54
#BLASFloats = (Float32, Float64, ComplexF32, ComplexF64)
65
BLASFloats = (Float32, ComplexF64) # full suite is too expensive on CI
@@ -10,12 +9,10 @@ using .TestSuite
109

1110
is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1211

13-
rng = StableRNG(12345)
14-
1512
m = 19
1613
for T in BLASFloats, n in (17, m, 23)
1714
TestSuite.seed_rng!(123)
1815
if !is_buildkite # doesn't work on GPU
19-
TestSuite.test_chainrules(T, (m, n), rng; atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
16+
TestSuite.test_chainrules(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
2017
end
2118
end

test/mooncake.jl

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
using MatrixAlgebraKit
22
using Test
3-
using StableRNGs
43
using CUDA, AMDGPU
54

65
#BLASFloats = (Float32, Float64, ComplexF32, ComplexF64)
@@ -11,18 +10,17 @@ using .TestSuite
1110

1211
is_buildkite = get(ENV, "BUILDKITE", "false") == "true"
1312

14-
rng = StableRNG(12345)
1513
m = 19
1614
for T in BLASFloats, n in (17, m, 23)
1715
TestSuite.seed_rng!(123)
1816
if is_buildkite
1917
if CUDA.functional()
20-
TestSuite.test_mooncake(CuMatrix{T}, (m, n), rng; atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
18+
TestSuite.test_mooncake(CuMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
2119
end
2220
#=if AMDGPU.functional()
23-
TestSuite.test_mooncake(ROCMatrix{T}, (m, n), rng; atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
21+
TestSuite.test_mooncake(ROCMatrix{T}, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
2422
end=# # not yet supported
2523
else
26-
TestSuite.test_mooncake(T, (m, n), rng; atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
24+
TestSuite.test_mooncake(T, (m, n); atol = m * n * TestSuite.precision(T), rtol = m * n * TestSuite.precision(T))
2725
end
2826
end

0 commit comments

Comments
 (0)