@@ -24,13 +24,13 @@ b = ArrayPartition(([0.0f0] |> cu, [0.0f0] |> cu, [0.0f0] |> cu))
2424
2525# Test adapt from ArrayPartition with CuArrays to ArrayPartition with CPU arrays
2626
27- a = CuArray (Float64 .([1. , 2. , 3. , 4. ]))
28- b = CuArray (Float64 .([1. , 2. , 3. , 4. ]))
27+ a = CuArray (Float64 .([1.0 , 2.0 , 3.0 , 4.0 ]))
28+ b = CuArray (Float64 .([1.0 , 2.0 , 3.0 , 4.0 ]))
2929part_a_gpu = ArrayPartition (a, b)
3030part_a = adapt (Array{Float32}, part_a_gpu)
3131
32- c = Float32 .([1. , 2. , 3. , 4. ])
33- d = Float32 .([1. , 2. , 3. , 4. ])
32+ c = Float32 .([1.0 , 2.0 , 3.0 , 4.0 ])
33+ d = Float32 .([1.0 , 2.0 , 3.0 , 4.0 ])
3434part_b = ArrayPartition (c, d)
3535
3636@test part_a == part_b # Test equality
@@ -42,7 +42,7 @@ for i in 1:length(part_a.x)
4242 @test typeof (sub_a) === typeof (sub_b) # Test type equality
4343end
4444
45- x = ArrayPartition ((CUDA. zeros (2 ),CUDA. zeros (2 )))
45+ x = ArrayPartition ((CUDA. zeros (2 ), CUDA. zeros (2 )))
4646@test ArrayInterface. zeromatrix (x) isa CuMatrix
47- @test size (ArrayInterface. zeromatrix (x)) == (4 ,4 )
48- @test maximum (abs, x) == 0f0
47+ @test size (ArrayInterface. zeromatrix (x)) == (4 , 4 )
48+ @test maximum (abs, x) == 0. 0f0
0 commit comments