Skip to content

Commit a428136

Browse files
committed
Test svd_trunc for GPU
1 parent 4f5bcb1 commit a428136

2 files changed

Lines changed: 12 additions & 12 deletions

File tree

test/svd.jl

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,28 +17,28 @@ for T in (BLASFloats..., GenericFloats...), m in (0, 54), n in (0, 37, m, 63)
1717
TestSuite.seed_rng!(123)
1818
if T BLASFloats
1919
if CUDA.functional()
20-
TestSuite.test_svd(CuMatrix{T}, (m, n); test_trunc = false)
20+
TestSuite.test_svd(CuMatrix{T}, (m, n))
2121
CUDA_SVD_ALGS = (
2222
CUSOLVER_QRIteration(),
2323
CUSOLVER_SVDPolar(),
2424
CUSOLVER_Jacobi(),
2525
)
26-
TestSuite.test_svd_algs(CuMatrix{T}, (m, n), CUDA_SVD_ALGS; test_trunc = false)
26+
TestSuite.test_svd_algs(CuMatrix{T}, (m, n), CUDA_SVD_ALGS)
2727
if n == m
28-
TestSuite.test_svd(Diagonal{T, CuVector{T}}, m; test_trunc = false)
29-
TestSuite.test_svd_algs(Diagonal{T, CuVector{T}}, m, (DiagonalAlgorithm(),); test_trunc = false)
28+
TestSuite.test_svd(Diagonal{T, CuVector{T}}, m)
29+
TestSuite.test_svd_algs(Diagonal{T, CuVector{T}}, m, (DiagonalAlgorithm(),))
3030
end
3131
end
3232
if AMDGPU.functional()
33-
TestSuite.test_svd(ROCMatrix{T}, (m, n); test_trunc = false)
33+
TestSuite.test_svd(ROCMatrix{T}, (m, n))
3434
AMD_SVD_ALGS = (
3535
ROCSOLVER_QRIteration(),
3636
ROCSOLVER_Jacobi(),
3737
)
38-
TestSuite.test_svd_algs(ROCMatrix{T}, (m, n), AMD_SVD_ALGS; test_trunc = false)
38+
TestSuite.test_svd_algs(ROCMatrix{T}, (m, n), AMD_SVD_ALGS)
3939
if n == m
40-
TestSuite.test_svd(Diagonal{T, ROCVector{T}}, m; test_trunc = false)
41-
TestSuite.test_svd_algs(Diagonal{T, ROCVector{T}}, m, (DiagonalAlgorithm(),); test_trunc = false)
40+
TestSuite.test_svd(Diagonal{T, ROCVector{T}}, m)
41+
TestSuite.test_svd_algs(Diagonal{T, ROCVector{T}}, m, (DiagonalAlgorithm(),))
4242
end
4343
end
4444
end

test/testsuite/svd.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -2,21 +2,21 @@ using TestExtras
22
using GenericLinearAlgebra
33
using LinearAlgebra: opnorm
44

5-
function test_svd(T::Type, sz; test_trunc = true, kwargs...)
5+
function test_svd(T::Type, sz; kwargs...)
66
summary_str = testargs_summary(T, sz)
77
return @testset "svd $summary_str" begin
88
test_svd_compact(T, sz; kwargs...)
99
test_svd_full(T, sz; kwargs...)
10-
test_trunc && test_svd_trunc(T, sz; kwargs...)
10+
test_svd_trunc(T, sz; kwargs...)
1111
end
1212
end
1313

14-
function test_svd_algs(T::Type, sz, algs; test_trunc = true, kwargs...)
14+
function test_svd_algs(T::Type, sz, algs; kwargs...)
1515
summary_str = testargs_summary(T, sz)
1616
return @testset "svd algorithms $summary_str" begin
1717
test_svd_compact_algs(T, sz, algs; kwargs...)
1818
test_svd_full_algs(T, sz, algs; kwargs...)
19-
test_trunc && test_svd_trunc_algs(T, sz, algs; kwargs...)
19+
test_svd_trunc_algs(T, sz, algs; kwargs...)
2020
end
2121
end
2222

0 commit comments

Comments
 (0)