Skip to content

Commit 8c1037b

Browse files
authored
Make lazy wrapper tests for GPUs more consistent
1 parent 612e56f commit 8c1037b

File tree

2 files changed

+6
-8
lines changed

2 files changed

+6
-8
lines changed

test/gpu/amdgpu.jl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,8 +6,6 @@ using LinearOperators, AMDGPU
66
B = ROCArray(rand(Float32, 10, 10))
77
C = ROCArray(rand(Float32, 20, 20))
88
M = BlockDiagonalOperator(A, B, C)
9-
v = ROCArray(rand(5))
10-
119

1210
v = ROCArray(rand(Float32, 35))
1311
y = M * v

test/gpu/nvidia.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -8,17 +8,17 @@ using LinearOperators, CUDA, CUDA.CUSPARSE, CUDA.CUSOLVER
88
A = CUDA.rand(5, 5)
99
B = CUDA.rand(10, 10)
1010
C = CUDA.rand(20, 20)
11-
M = BlockDiagonalOperator(A, B, C)
12-
v = CUDA.rand(5)
11+
M = BlockDiagonalOperator(A, B, C)
12+
13+
v = CUDA.rand(35)
14+
y = M * v
15+
@test y isa CuVector{Float32}
1316

1417
@test LinearOperators.storage_type(A) == LinearOperators.storage_type(adjoint(A))
1518
@test LinearOperators.storage_type(A) == LinearOperators.storage_type(transpose(A))
1619
@test LinearOperators.storage_type(A) == LinearOperators.storage_type(adjoint(A))
1720
@test LinearOperators.storage_type(Diagonal(v)) == typeof(v)
18-
1921

20-
v = CUDA.rand(35)
21-
y = M * v
22-
@test y isa CuVector{Float32}
22+
2323
@testset "Nvidia S kwarg" test_S_kwarg(arrayType = CuArray)
2424
end

0 commit comments

Comments
 (0)