Skip to content

Commit 54de102

Browse files
authored
Merge pull request #211 from JuliaGPU/ksh/test
Some tests
2 parents 5dd7750 + 7a41bb0 commit 54de102

File tree

1 file changed

+14
-1
lines changed

1 file changed

+14
-1
lines changed

src/testsuite/linalg.jl

Lines changed: 14 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,8 +4,21 @@ function test_linalg(AT)
44
@test compare(adjoint, AT, rand(Float32, 32, 32))
55
@test compare(transpose, AT, rand(Float32, 32, 32))
66
@test compare(transpose!, AT, Array{Float32}(undef, 32, 32), rand(Float32, 32, 32))
7+
@test compare(transpose!, AT, Array{Float32}(undef, 128, 32), rand(Float32, 32, 128))
8+
end
9+
10+
@testset "copyto! for triangular" begin
11+
ga = Array{Float32}(undef, 128, 128)
12+
gb = AT{Float32}(undef, 128, 128)
13+
rand!(gb)
14+
copyto!(ga, UpperTriangular(gb))
15+
@test ga == Array(collect(UpperTriangular(gb)))
16+
ga = Array{Float32}(undef, 128, 128)
17+
gb = AT{Float32}(undef, 128, 128)
18+
rand!(gb)
19+
copyto!(ga, LowerTriangular(gb))
20+
@test ga == Array(collect(LowerTriangular(gb)))
721
end
8-
922
@testset "permutedims" begin
1023
@test compare(x -> permutedims(x, (2, 1)), AT, rand(Float32, 2, 3))
1124
@test compare(x -> permutedims(x, (2, 1, 3)), AT, rand(Float32, 4, 5, 6))

0 commit comments

Comments
 (0)