Skip to content

Commit ce8bb40

Browse files
traktofonandreasnoack
authored andcommitted
Fix more deprecations.
Note: (map)reducedim probably currently falls back to the method for AbstractArray.
1 parent 0b7af3c commit ce8bb40

File tree

3 files changed

+45
-62
lines changed

3 files changed

+45
-62
lines changed

src/linalg.jl

Lines changed: 9 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,14 @@
11
function Base.copy(D::Adjoint{T,<:DArray{T,2}}) where T
22
DArray(reverse(size(D)), procs(D)) do I
3-
lp = Array{T}(map(length, I))
3+
lp = Array{T}(undef, map(length, I))
44
rp = convert(Array, D[reverse(I)...])
55
adjoint!(lp, rp)
66
end
77
end
88

99
function Base.copy(D::Transpose{T,<:DArray{T,2}}) where T
1010
DArray(reverse(size(D)), procs(D)) do I
11-
lp = Array{T}(map(length, I))
11+
lp = Array{T}(undef, map(length, I))
1212
rp = convert(Array, D[reverse(I)...])
1313
transpose!(lp, rp)
1414
end
@@ -94,7 +94,7 @@ function A_mul_B!(α::Number, A::DMatrix, x::AbstractVector, β::Number, y::DVec
9494
end
9595

9696
# Multiply on each tile of A
97-
R = Array{Future}(size(A.pids)...)
97+
R = Array{Future}(undef, size(A.pids))
9898
for j = 1:size(A.pids, 2)
9999
xj = x[A.cuts[2][j]:A.cuts[2][j + 1] - 1]
100100
for i = 1:size(A.pids, 1)
@@ -138,7 +138,7 @@ function Ac_mul_B!(α::Number, A::DMatrix, x::AbstractVector, β::Number, y::DVe
138138
end
139139

140140
# Multiply on each tile of A
141-
R = Array{Future}(reverse(size(A.pids))...)
141+
R = Array{Future}(undef, reverse(size(A.pids)))
142142
for j = 1:size(A.pids, 1)
143143
xj = x[A.cuts[1][j]:A.cuts[1][j + 1] - 1]
144144
for i = 1:size(A.pids, 2)
@@ -206,9 +206,9 @@ function _matmatmul!(α::Number, A::DMatrix, B::AbstractMatrix, β::Number, C::D
206206

207207
# Multiply on each tile of A
208208
if tA == 'N'
209-
R = Array{Future}(size(procs(A))..., size(procs(C), 2))
209+
R = Array{Future}(undef, size(procs(A))..., size(procs(C), 2))
210210
else
211-
R = Array{Future}(reverse(size(procs(A)))..., size(procs(C), 2))
211+
R = Array{Future}(undef, reverse(size(procs(A)))..., size(procs(C), 2))
212212
end
213213
for j = 1:size(A.pids, Ad2)
214214
for k = 1:size(C.pids, 2)
@@ -221,7 +221,7 @@ function _matmatmul!(α::Number, A::DMatrix, B::AbstractMatrix, β::Number, C::D
221221
if tA == 'T'
222222
return transpose(localpart(A))*convert(localtype(B), Bjk)
223223
elseif tA == 'C'
224-
return ctranspose(localpart(A))*convert(localtype(B), Bjk)
224+
return adjoint(localpart(A))*convert(localtype(B), Bjk)
225225
else
226226
return localpart(A)*convert(localtype(B), Bjk)
227227
end
@@ -261,12 +261,12 @@ At_mul_B!(C::DMatrix, A::DMatrix, B::AbstractMatrix) = At_mul_B!(one(eltype(C)),
261261

262262
_matmul_op = (t,s) -> t*s + t*s
263263

264-
function (*)(A::DMatrix, x::AbstractVector)
264+
function Base.:*(A::DMatrix, x::AbstractVector)
265265
T = Base.promote_op(_matmul_op, eltype(A), eltype(x))
266266
y = DArray(I -> Array{T}(map(length, I)), (size(A, 1),), procs(A)[:,1], (size(procs(A), 1),))
267267
return A_mul_B!(one(T), A, x, zero(T), y)
268268
end
269-
function (*)(A::DMatrix, B::AbstractMatrix)
269+
function Base.:*(A::DMatrix, B::AbstractMatrix)
270270
T = Base.promote_op(_matmul_op, eltype(A), eltype(B))
271271
C = DArray(I -> Array{T}(map(length, I)),
272272
(size(A, 1), size(B, 2)),

src/mapreduce.jl

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -127,10 +127,6 @@ function Base.mapreducedim!(f, op, R::DArray, A::DArray)
127127
return mapreducedim_between!(identity, op, R, B, region)
128128
end
129129

130-
# function Base.mapreducedim(f, op, R::DArray, A::DArray)
131-
# Base.mapreducedim!(f, op, Base.reducedim_initarray(A, region, v0), A)
132-
# end
133-
134130
function nnz(A::DArray)
135131
B = asyncmap(A.pids) do p
136132
remotecall_fetch(nnzlocalpart, p, A)

test/darray.jl

Lines changed: 36 additions & 49 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,6 @@
1-
<<<<<<< HEAD
21
using Test, LinearAlgebra, SpecialFunctions
3-
=======
4-
using SpecialFunctions
5-
using LinearAlgebra
6-
>>>>>>> scale! -> [lr]mul!
2+
using Statistics: mean
3+
@everywhere using SparseArrays: sprandn
74

85
@testset "test distribute and other constructors" begin
96
A = rand(1:100, (100,100))
@@ -210,29 +207,17 @@ end
210207

211208
check_leaks()
212209

213-
<<<<<<< HEAD
214210
@testset "test rmul!(Diagonal, A)" begin
215-
=======
216-
@testset "test [lr]mul!(b, A)" begin
217-
>>>>>>> scale! -> [lr]mul!
218211
A = randn(100, 100)
219212
b = randn(100)
220213
D = Diagonal(b)
221214
DA = distribute(A)
222-
<<<<<<< HEAD
223215
@test lmul!(D, A) == lmul!(D, DA)
224-
=======
225-
@test lmul!(Diagonal(b), A) == lmul!(Diagonal(b), DA)
226-
>>>>>>> scale! -> [lr]mul!
227216
close(DA)
228217
A = randn(100, 100)
229218
b = randn(100)
230219
DA = distribute(A)
231-
<<<<<<< HEAD
232220
@test rmul!(A, D) == rmul!(DA, D)
233-
=======
234-
@test rmul!(A, Diagonal(b)) == rmul!(A, Diagonal(b))
235-
>>>>>>> scale! -> [lr]mul!
236221
close(DA)
237222
end
238223

@@ -242,6 +227,7 @@ check_leaks()
242227
for _ = 1:25, f = [x -> Int128(2x), x -> Int128(x^2), x -> Int128(x^2 + 2x - 1)], opt = [+, *]
243228
A = rand(1:5, rand(2:30))
244229
DA = distribute(A)
230+
@test DA isa DArray
245231
@test mapreduce(f, opt, DA) - mapreduce(f, opt, A) == 0
246232
close(DA)
247233
end
@@ -252,15 +238,16 @@ check_leaks()
252238
@testset "test mapreducedim on DArrays" begin
253239
D = DArray(I->fill(myid(), map(length,I)), (73,73), [MYID, OTHERIDS])
254240
D2 = map(x->1, D)
255-
@test mapreducedim(t -> t*t, +, D2, 1) == mapreducedim(t -> t*t, +, convert(Array, D2), 1)
256-
@test mapreducedim(t -> t*t, +, D2, 2) == mapreducedim(t -> t*t, +, convert(Array, D2), 2)
257-
@test mapreducedim(t -> t*t, +, D2, (1,2)) == mapreducedim(t -> t*t, +, convert(Array, D2), (1,2))
241+
@test D2 isa DArray
242+
@test mapreduce(t -> t*t, +, D2, dims=1) == mapreduce(t -> t*t, +, convert(Array, D2), dims=1)
243+
@test mapreduce(t -> t*t, +, D2, dims=2) == mapreduce(t -> t*t, +, convert(Array, D2), dims=2)
244+
@test mapreduce(t -> t*t, +, D2, dims=(1,2)) == mapreduce(t -> t*t, +, convert(Array, D2), dims=(1,2))
258245

259246
# Test non-regularly chunked DArrays
260247
r1 = DistributedArrays.remotecall(() -> sprandn(3, 10, 0.1), workers()[1])
261248
r2 = DistributedArrays.remotecall(() -> sprandn(7, 10, 0.1), workers()[2])
262249
D = DArray(reshape([r1; r2], (2,1)))
263-
@test Array(sum(D, 2)) == sum(Array(D), 2)
250+
@test Array(sum(D, dims=2)) == sum(Array(D), dims=2)
264251

265252
# close(D)
266253
# close(D2)
@@ -275,10 +262,10 @@ check_leaks()
275262
A = convert(Array, DA)
276263

277264
@testset "dimension $dms" for dms in (1, 2, 3, (1,2), (1,3), (2,3), (1,2,3))
278-
@test mapreducedim(t -> t*t, +, A, dms) mapreducedim(t -> t*t, +, DA, dms)
279-
@test mapreducedim(t -> t*t, +, A, dms, 1.0) mapreducedim(t -> t*t, +, DA, dms, 1.0)
280-
@test reducedim(*, A, dms) reducedim(*, DA, dms)
281-
@test reducedim(*, A, dms, 2.0) reducedim(*, DA, dms, 2.0)
265+
@test mapreduce(t -> t*t, +, A, dims=dms) mapreduce(t -> t*t, +, DA, dims=dms)
266+
@test mapreduce(t -> t*t, +, A, dims=dms, init=1.0) mapreduce(t -> t*t, +, DA, dims=dms, init=1.0)
267+
@test reduce(*, A, dims=dms) reduce(*, DA, dims=dms)
268+
@test reduce(*, A, dims=dms, init=2.0) reduce(*, DA, dims=dms, init=2.0)
282269
end
283270
close(DA)
284271
d_closeall() # temp created by the mapreduce above
@@ -293,7 +280,7 @@ check_leaks()
293280

294281
@testset "test $f for dimension $dms" for f in (mean, ), dms in (1, 2, 3, (1,2), (1,3), (2,3), (1,2,3))
295282
# std is pending implementation
296-
@test f(DA,dms) f(A,dms)
283+
@test f(DA, dims=dms) f(A, dims=dms)
297284
end
298285

299286
close(DA)
@@ -308,7 +295,7 @@ check_leaks()
308295

309296
# sum either throws an ArgumentError or a CompositeException of ArgumentErrors
310297
try
311-
sum(DA, -1)
298+
sum(DA, dims=-1)
312299
catch err
313300
if isa(err, CompositeException)
314301
@test !isempty(err.exceptions)
@@ -322,7 +309,7 @@ check_leaks()
322309
end
323310
end
324311
try
325-
sum(DA, 0)
312+
sum(DA, dims=0)
326313
catch err
327314
if isa(err, CompositeException)
328315
@test !isempty(err.exceptions)
@@ -337,9 +324,9 @@ check_leaks()
337324
end
338325

339326
@test sum(DA) sum(A)
340-
@test sum(DA,1) sum(A,1)
341-
@test sum(DA,2) sum(A,2)
342-
@test sum(DA,3) sum(A,3)
327+
@test sum(DA, dims=1) sum(A, dims=1)
328+
@test sum(DA, dims=2) sum(A, dims=2)
329+
@test sum(DA, dims=3) sum(A, dims=3)
343330
close(DA)
344331
d_closeall() # temporaries created above
345332
end
@@ -360,7 +347,7 @@ end
360347

361348
check_leaks()
362349

363-
# test length / endof
350+
# test length / lastindex
364351
@testset "test collections API" begin
365352
A = randn(23,23)
366353
DA = distribute(A)
@@ -369,16 +356,16 @@ check_leaks()
369356
@test length(DA) == length(A)
370357
end
371358

372-
@testset "test endof" begin
373-
@test endof(DA) == endof(A)
359+
@testset "test lastindex" begin
360+
@test lastindex(DA) == lastindex(A)
374361
end
375362
close(DA)
376363
end
377364

378365
check_leaks()
379366

380367
@testset "test max / min / sum" begin
381-
a = map(x -> Int(round(rand() * 100)) - 50, Array{Int}(100,1000))
368+
a = map(x -> Int(round(rand() * 100)) - 50, Array{Int}(undef,100,1000))
382369
d = distribute(a)
383370

384371
@test sum(d) == sum(a)
@@ -395,7 +382,7 @@ end
395382
check_leaks()
396383

397384
@testset "test all / any" begin
398-
a = map(x->Int(round(rand() * 100)) - 50, Array{Int}(100,1000))
385+
a = map(x->Int(round(rand() * 100)) - 50, Array{Int}(undef,100,1000))
399386
a = [true for i in 1:100]
400387
d = distribute(a)
401388

@@ -651,25 +638,25 @@ end
651638

652639
check_leaks()
653640

654-
@testset "test c/transpose" begin
655-
@testset "test ctranspose real" begin
641+
@testset "test transpose/adjoint" begin
642+
@testset "test transpose real" begin
656643
A = drand(Float64, 100, 200)
657-
@test A' == Array(A)'
644+
@test transpose(A) == transpose(Array(A))
658645
close(A)
659646
end
660-
@testset "test ctranspose complex" begin
661-
A = drand(Complex128, 200, 100)
662-
@test A' == Array(A)'
647+
@testset "test transpose complex" begin
648+
A = drand(ComplexF64, 200, 100)
649+
@test transpose(A) == transpose(Array(A))
663650
close(A)
664651
end
665-
@testset "test transpose real" begin
652+
@testset "test adjoint real" begin
666653
A = drand(Float64, 200, 100)
667-
@test transpose(A) == transpose(Array(A))
654+
@test adjoint(A) == adjoint(Array(A))
668655
close(A)
669656
end
670-
@testset "test ctranspose complex" begin
671-
A = drand(Complex128, 100, 200)
672-
@test transpose(A) == transpose(Array(A))
657+
@testset "test adjoint complex" begin
658+
A = drand(ComplexF64, 100, 200)
659+
@test adjoint(A) == adjoint(Array(A))
673660
close(A)
674661
end
675662

@@ -712,8 +699,8 @@ check_leaks()
712699
sinh, sinpi, sqrt, tan, tand, tanh, trigamma)
713700
@test f.(a) == f.(b)
714701
end
715-
a = a + 1
716-
b = b + 1
702+
a = a .+ 1
703+
b = b .+ 1
717704
@testset "$f" for f in (asec, asecd, acosh, acsc, acscd, acoth)
718705
@test f.(a) == f.(b)
719706
end

0 commit comments

Comments
 (0)