#37433

pending completion

Build # #37433

Build Type

push

local

Committed by

web-flow

Commit Message

Merge pull request #48513 from JuliaLang/jn/extend-once

ensure extension triggers are only run by the package that satified them

Run Details

60 of 60 new or added lines in 1 file covered. (100.0%)

72324 of 82360 relevant lines covered (87.81%)

31376331.4 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

95.05

/base/permuteddimsarray.jl

# This file is a part of Julia. License is MIT: https://julialang.org/license

module PermutedDimsArrays

import Base: permutedims, permutedims!
export PermutedDimsArray

# Some day we will want storage-order-aware iteration, so put perm in the parameters
struct PermutedDimsArray{T,N,perm,iperm,AA<:AbstractArray} <: AbstractArray{T,N}
    parent::AA

    function PermutedDimsArray{T,N,perm,iperm,AA}(data::AA) where {T,N,perm,iperm,AA<:AbstractArray}
        (isa(perm, NTuple{N,Int}) && isa(iperm, NTuple{N,Int})) || error("perm and iperm must both be NTuple{$N,Int}")
        isperm(perm) || throw(ArgumentError(string(perm, " is not a valid permutation of dimensions 1:", N)))
        all(map(d->iperm[perm[d]]==d, 1:N)) || throw(ArgumentError(string(perm, " and ", iperm, " must be inverses")))
        new(data)
    end
end

"""
    PermutedDimsArray(A, perm) -> B

Given an AbstractArray `A`, create a view `B` such that the
dimensions appear to be permuted. Similar to `permutedims`, except
that no copying occurs (`B` shares storage with `A`).

See also [`permutedims`](@ref), [`invperm`](@ref).

# Examples
```jldoctest
julia> A = rand(3,5,4);

julia> B = PermutedDimsArray(A, (3,1,2));

julia> size(B)
(4, 3, 5)

julia> B[3,1,2] == A[1,2,3]
true
```
"""
function PermutedDimsArray(data::AbstractArray{T,N}, perm) where {T,N}
    length(perm) == N || throw(ArgumentError(string(perm, " is not a valid permutation of dimensions 1:", N)))
    iperm = invperm(perm)
    PermutedDimsArray{T,N,(perm...,),(iperm...,),typeof(data)}(data)
end

Base.parent(A::PermutedDimsArray) = A.parent
Base.size(A::PermutedDimsArray{T,N,perm}) where {T,N,perm} = genperm(size(parent(A)), perm)
Base.axes(A::PermutedDimsArray{T,N,perm}) where {T,N,perm} = genperm(axes(parent(A)), perm)
Base.has_offset_axes(A::PermutedDimsArray) = Base.has_offset_axes(A.parent)

Base.similar(A::PermutedDimsArray, T::Type, dims::Base.Dims) = similar(parent(A), T, dims)

Base.unsafe_convert(::Type{Ptr{T}}, A::PermutedDimsArray{T}) where {T} = Base.unsafe_convert(Ptr{T}, parent(A))

# It's OK to return a pointer to the first element, and indeed quite
# useful for wrapping C routines that require a different storage
# order than used by Julia. But for an array with unconventional
# storage order, a linear offset is ambiguous---is it a memory offset
# or a linear index?
Base.pointer(A::PermutedDimsArray, i::Integer) = throw(ArgumentError("pointer(A, i) is deliberately unsupported for PermutedDimsArray"))

function Base.strides(A::PermutedDimsArray{T,N,perm}) where {T,N,perm}
    s = strides(parent(A))
    ntuple(d->s[perm[d]], Val(N))
end
Base.elsize(::Type{<:PermutedDimsArray{<:Any, <:Any, <:Any, <:Any, P}}) where {P} = Base.elsize(P)

@inline function Base.getindex(A::PermutedDimsArray{T,N,perm,iperm}, I::Vararg{Int,N}) where {T,N,perm,iperm}
    @boundscheck checkbounds(A, I...)
    @inbounds val = getindex(A.parent, genperm(I, iperm)...)
    val
end
@inline function Base.setindex!(A::PermutedDimsArray{T,N,perm,iperm}, val, I::Vararg{Int,N}) where {T,N,perm,iperm}
    @boundscheck checkbounds(A, I...)
    @inbounds setindex!(A.parent, val, genperm(I, iperm)...)
    val
end

@inline genperm(I::NTuple{N,Any}, perm::Dims{N}) where {N} = ntuple(d -> I[perm[d]], Val(N))
@inline genperm(I, perm::AbstractVector{Int}) = genperm(I, (perm...,))

"""
    permutedims(A::AbstractArray, perm)

Permute the dimensions of array `A`. `perm` is a vector or a tuple of length `ndims(A)`
specifying the permutation.

See also [`permutedims!`](@ref), [`PermutedDimsArray`](@ref), [`transpose`](@ref), [`invperm`](@ref).

# Examples
```jldoctest
julia> A = reshape(Vector(1:8), (2,2,2))
2×2×2 Array{Int64, 3}:
[:, :, 1] =
 1  3
 2  4

[:, :, 2] =
 5  7
 6  8

julia> perm = (3, 1, 2); # put the last dimension first

julia> B = permutedims(A, perm)
2×2×2 Array{Int64, 3}:
[:, :, 1] =
 1  2
 5  6

[:, :, 2] =
 3  4
 7  8

julia> A == permutedims(B, invperm(perm)) # the inverse permutation
true
```

For each dimension `i` of `B = permutedims(A, perm)`, its corresponding dimension of `A`
will be `perm[i]`. This means the equality `size(B, i) == size(A, perm[i])` holds.

```jldoctest
julia> A = randn(5, 7, 11, 13);

julia> perm = [4, 1, 3, 2];

julia> B = permutedims(A, perm);

julia> size(B)
(13, 5, 11, 7)

julia> size(A)[perm] == ans
true
```
"""
function permutedims(A::AbstractArray, perm)
    dest = similar(A, genperm(axes(A), perm))
    permutedims!(dest, A, perm)
end

"""
    permutedims(m::AbstractMatrix)

Permute the dimensions of the matrix `m`, by flipping the elements across the diagonal of
the matrix. Differs from `LinearAlgebra`'s [`transpose`](@ref) in that the
operation is not recursive.

# Examples
```jldoctest; setup = :(using LinearAlgebra)
julia> a = [1 2; 3 4];

julia> b = [5 6; 7 8];

julia> c = [9 10; 11 12];

julia> d = [13 14; 15 16];

julia> X = [[a] [b]; [c] [d]]
2×2 Matrix{Matrix{Int64}}:
 [1 2; 3 4]     [5 6; 7 8]
 [9 10; 11 12]  [13 14; 15 16]

julia> permutedims(X)
2×2 Matrix{Matrix{Int64}}:
 [1 2; 3 4]  [9 10; 11 12]
 [5 6; 7 8]  [13 14; 15 16]

julia> transpose(X)
2×2 transpose(::Matrix{Matrix{Int64}}) with eltype Transpose{Int64, Matrix{Int64}}:
 [1 3; 2 4]  [9 11; 10 12]
 [5 7; 6 8]  [13 15; 14 16]
```
"""
permutedims(A::AbstractMatrix) = permutedims(A, (2,1))

"""
    permutedims(v::AbstractVector)

Reshape vector `v` into a `1 × length(v)` row matrix.
Differs from `LinearAlgebra`'s [`transpose`](@ref) in that
the operation is not recursive.

# Examples
```jldoctest; setup = :(using LinearAlgebra)
julia> permutedims([1, 2, 3, 4])
1×4 Matrix{Int64}:
 1  2  3  4

julia> V = [[[1 2; 3 4]]; [[5 6; 7 8]]]
2-element Vector{Matrix{Int64}}:
 [1 2; 3 4]
 [5 6; 7 8]

julia> permutedims(V)
1×2 Matrix{Matrix{Int64}}:
 [1 2; 3 4]  [5 6; 7 8]

julia> transpose(V)
1×2 transpose(::Vector{Matrix{Int64}}) with eltype Transpose{Int64, Matrix{Int64}}:
 [1 3; 2 4]  [5 7; 6 8]
```
"""
permutedims(v::AbstractVector) = reshape(v, (1, length(v)))

"""
    permutedims!(dest, src, perm)

Permute the dimensions of array `src` and store the result in the array `dest`. `perm` is a
vector specifying a permutation of length `ndims(src)`. The preallocated array `dest` should
have `size(dest) == size(src)[perm]` and is completely overwritten. No in-place permutation
is supported and unexpected results will happen if `src` and `dest` have overlapping memory
regions.

See also [`permutedims`](@ref).
"""
function permutedims!(dest, src::AbstractArray, perm)
    Base.checkdims_perm(dest, src, perm)
    P = PermutedDimsArray(dest, invperm(perm))
    _copy!(P, src)
    return dest
end

function Base.copyto!(dest::PermutedDimsArray{T,N}, src::AbstractArray{T,N}) where {T,N}
    checkbounds(dest, axes(src)...)
    _copy!(dest, src)
end
Base.copyto!(dest::PermutedDimsArray, src::AbstractArray) = _copy!(dest, src)

function _copy!(P::PermutedDimsArray{T,N,perm}, src) where {T,N,perm}
    # If dest/src are "close to dense," then it pays to be cache-friendly.
    # Determine the first permuted dimension
    d = 0  # d+1 will hold the first permuted dimension of src
    while d < ndims(src) && perm[d+1] == d+1
        d += 1
    end
    if d == ndims(src)
        copyto!(parent(P), src) # it's not permuted
    else
        R1 = CartesianIndices(axes(src)[1:d])
        d1 = findfirst(isequal(d+1), perm)::Int  # first permuted dim of dest
        R2 = CartesianIndices(axes(src)[d+2:d1-1])
        R3 = CartesianIndices(axes(src)[d1+1:end])
        _permutedims!(P, src, R1, R2, R3, d+1, d1)
    end
    return P
end

@noinline function _permutedims!(P::PermutedDimsArray, src, R1::CartesianIndices{0}, R2, R3, ds, dp)
    ip, is = axes(src, dp), axes(src, ds)
    for jo in first(ip):8:last(ip), io in first(is):8:last(is)
        for I3 in R3, I2 in R2
            for j in jo:min(jo+7, last(ip))
                for i in io:min(io+7, last(is))
                    @inbounds P[i, I2, j, I3] = src[i, I2, j, I3]
                end
            end
        end
    end
    P
end

@noinline function _permutedims!(P::PermutedDimsArray, src, R1, R2, R3, ds, dp)
    ip, is = axes(src, dp), axes(src, ds)
    for jo in first(ip):8:last(ip), io in first(is):8:last(is)
        for I3 in R3, I2 in R2
            for j in jo:min(jo+7, last(ip))
                for i in io:min(io+7, last(is))
                    for I1 in R1
                        @inbounds P[I1, i, I2, j, I3] = src[I1, i, I2, j, I3]
                    end
                end
            end
        end
    end
    P
end

const CommutativeOps = Union{typeof(+),typeof(Base.add_sum),typeof(min),typeof(max),typeof(Base._extrema_rf),typeof(|),typeof(&)}

function Base._mapreduce_dim(f, op::CommutativeOps, init::Base._InitialValue, A::PermutedDimsArray, dims::Colon)
    Base._mapreduce_dim(f, op, init, parent(A), dims)
end
function Base._mapreduce_dim(f::typeof(identity), op::Union{typeof(Base.mul_prod),typeof(*)}, init::Base._InitialValue, A::PermutedDimsArray{<:Union{Real,Complex}}, dims::Colon)
    Base._mapreduce_dim(f, op, init, parent(A), dims)
end

function Base.mapreducedim!(f, op::CommutativeOps, B::AbstractArray{T,N}, A::PermutedDimsArray{S,N,perm,iperm}) where {T,S,N,perm,iperm}
    C = PermutedDimsArray{T,N,iperm,perm,typeof(B)}(B) # make the inverse permutation for the output
    Base.mapreducedim!(f, op, C, parent(A))
    B
end
function Base.mapreducedim!(f::typeof(identity), op::Union{typeof(Base.mul_prod),typeof(*)}, B::AbstractArray{T,N}, A::PermutedDimsArray{<:Union{Real,Complex},N,perm,iperm}) where {T,N,perm,iperm}
    C = PermutedDimsArray{T,N,iperm,perm,typeof(B)}(B) # make the inverse permutation for the output
    Base.mapreducedim!(f, op, C, parent(A))
    B
end

function Base.showarg(io::IO, A::PermutedDimsArray{T,N,perm}, toplevel) where {T,N,perm}
    print(io, "PermutedDimsArray(")
    Base.showarg(io, parent(A), false)
    print(io, ", ", perm, ')')
    toplevel && print(io, " with eltype ", eltype(A))
    return nothing
end

end

1	# This file is a part of Julia. License is MIT: https://julialang.org/license
2
3	module PermutedDimsArrays
4
5	import Base: permutedims, permutedims!
6	export PermutedDimsArray
7
8	# Some day we will want storage-order-aware iteration, so put perm in the parameters
9	struct PermutedDimsArray{T,N,perm,iperm,AA<:AbstractArray} <: AbstractArray{T,N}
10	parent::AA
11
12	function PermutedDimsArray{T,N,perm,iperm,AA}(data::AA) where {T,N,perm,iperm,AA<:AbstractArray}	332✔
13	(isa(perm, NTuple{N,Int}) && isa(iperm, NTuple{N,Int})) \|\| error("perm and iperm must both be NTuple{$N,Int}")	332✔
14	isperm(perm) \|\| throw(ArgumentError(string(perm, " is not a valid permutation of dimensions 1:", N)))	332✔
15	all(map(d->iperm[perm[d]]==d, 1:N)) \|\| throw(ArgumentError(string(perm, " and ", iperm, " must be inverses")))	1,036✔
16	new(data)	332✔
17	end
18	end
19
20	"""
21	PermutedDimsArray(A, perm) -> B
22
23	Given an AbstractArray `A`, create a view `B` such that the
24	dimensions appear to be permuted. Similar to `permutedims`, except
25	that no copying occurs (`B` shares storage with `A`).
26
27	See also [`permutedims`](@ref), [`invperm`](@ref).
28
29	# Examples
30	```jldoctest
31	julia> A = rand(3,5,4);
32
33	julia> B = PermutedDimsArray(A, (3,1,2));
34
35	julia> size(B)
36	(4, 3, 5)
37
38	julia> B[3,1,2] == A[1,2,3]
39	true
40	```
41	"""
42	function PermutedDimsArray(data::AbstractArray{T,N}, perm) where {T,N}	316✔
43	length(perm) == N \|\| throw(ArgumentError(string(perm, " is not a valid permutation of dimensions 1:", N)))	316✔
44	iperm = invperm(perm)	316✔
45	PermutedDimsArray{T,N,(perm...,),(iperm...,),typeof(data)}(data)	314✔
46	end
47
48	Base.parent(A::PermutedDimsArray) = A.parent	246,287✔
49	Base.size(A::PermutedDimsArray{T,N,perm}) where {T,N,perm} = genperm(size(parent(A)), perm)	49,693✔
50	Base.axes(A::PermutedDimsArray{T,N,perm}) where {T,N,perm} = genperm(axes(parent(A)), perm)	142,935✔
51	Base.has_offset_axes(A::PermutedDimsArray) = Base.has_offset_axes(A.parent)	×
52
53	Base.similar(A::PermutedDimsArray, T::Type, dims::Base.Dims) = similar(parent(A), T, dims)	23✔
54
55	Base.unsafe_convert(::Type{Ptr{T}}, A::PermutedDimsArray{T}) where {T} = Base.unsafe_convert(Ptr{T}, parent(A))	17,981✔
56
57	# It's OK to return a pointer to the first element, and indeed quite
58	# useful for wrapping C routines that require a different storage
59	# order than used by Julia. But for an array with unconventional
60	# storage order, a linear offset is ambiguous---is it a memory offset
61	# or a linear index?
62	Base.pointer(A::PermutedDimsArray, i::Integer) = throw(ArgumentError("pointer(A, i) is deliberately unsupported for PermutedDimsArray"))	81✔
63
64	function Base.strides(A::PermutedDimsArray{T,N,perm}) where {T,N,perm}	35,602✔
65	s = strides(parent(A))	35,602✔
66	ntuple(d->s[perm[d]], Val(N))	141,278✔
67	end
68	Base.elsize(::Type{<:PermutedDimsArray{<:Any, <:Any, <:Any, <:Any, P}}) where {P} = Base.elsize(P)	35,560✔
69
70	@inline function Base.getindex(A::PermutedDimsArray{T,N,perm,iperm}, I::Vararg{Int,N}) where {T,N,perm,iperm}	71,316✔
71	@boundscheck checkbounds(A, I...)	71,316✔
72	@inbounds val = getindex(A.parent, genperm(I, iperm)...)	71,316✔
73	val	71,316✔
74	end
75	@inline function Base.setindex!(A::PermutedDimsArray{T,N,perm,iperm}, val, I::Vararg{Int,N}) where {T,N,perm,iperm}	3,356✔
76	@boundscheck checkbounds(A, I...)	3,356✔
77	@inbounds setindex!(A.parent, val, genperm(I, iperm)...)	3,356✔
78	val	3,356✔
79	end
80
81	@inline genperm(I::NTuple{N,Any}, perm::Dims{N}) where {N} = ntuple(d -> I[perm[d]], Val(N))	1,067,388✔
82	@inline genperm(I, perm::AbstractVector{Int}) = genperm(I, (perm...,))	7✔
83
84	"""
85	permutedims(A::AbstractArray, perm)
86
87	Permute the dimensions of array `A`. `perm` is a vector or a tuple of length `ndims(A)`
88	specifying the permutation.
89
90	See also [`permutedims!`](@ref), [`PermutedDimsArray`](@ref), [`transpose`](@ref), [`invperm`](@ref).
91
92	# Examples
93	```jldoctest
94	julia> A = reshape(Vector(1:8), (2,2,2))
95	2×2×2 Array{Int64, 3}:
96	[:, :, 1] =
97	1 3
98	2 4
99
100	[:, :, 2] =
101	5 7
102	6 8
103
104	julia> perm = (3, 1, 2); # put the last dimension first
105
106	julia> B = permutedims(A, perm)
107	2×2×2 Array{Int64, 3}:
108	[:, :, 1] =
109	1 2
110	5 6
111
112	[:, :, 2] =
113	3 4
114	7 8
115
116	julia> A == permutedims(B, invperm(perm)) # the inverse permutation
117	true
118	```
119
120	For each dimension `i` of `B = permutedims(A, perm)`, its corresponding dimension of `A`
121	will be `perm[i]`. This means the equality `size(B, i) == size(A, perm[i])` holds.
122
123	```jldoctest
124	julia> A = randn(5, 7, 11, 13);
125
126	julia> perm = [4, 1, 3, 2];
127
128	julia> B = permutedims(A, perm);
129
130	julia> size(B)
131	(13, 5, 11, 7)
132
133	julia> size(A)[perm] == ans
134	true
135	```
136	"""
137	function permutedims(A::AbstractArray, perm)	14✔
138	dest = similar(A, genperm(axes(A), perm))	18✔
139	permutedims!(dest, A, perm)	14✔
140	end
141
142	"""
143	permutedims(m::AbstractMatrix)
144
145	Permute the dimensions of the matrix `m`, by flipping the elements across the diagonal of
146	the matrix. Differs from `LinearAlgebra`'s [`transpose`](@ref) in that the
147	operation is not recursive.
148
149	# Examples
150	```jldoctest; setup = :(using LinearAlgebra)
151	julia> a = [1 2; 3 4];
152
153	julia> b = [5 6; 7 8];
154
155	julia> c = [9 10; 11 12];
156
157	julia> d = [13 14; 15 16];
158
159	julia> X = [[a] [b]; [c] [d]]
160	2×2 Matrix{Matrix{Int64}}:
161	[1 2; 3 4] [5 6; 7 8]
162	[9 10; 11 12] [13 14; 15 16]
163
164	julia> permutedims(X)
165	2×2 Matrix{Matrix{Int64}}:
166	[1 2; 3 4] [9 10; 11 12]
167	[5 6; 7 8] [13 14; 15 16]
168
169	julia> transpose(X)
170	2×2 transpose(::Matrix{Matrix{Int64}}) with eltype Transpose{Int64, Matrix{Int64}}:
171	[1 3; 2 4] [9 11; 10 12]
172	[5 7; 6 8] [13 15; 14 16]
173	```
174	"""
175	permutedims(A::AbstractMatrix) = permutedims(A, (2,1))	4✔
176
177	"""
178	permutedims(v::AbstractVector)
179
180	Reshape vector `v` into a `1 × length(v)` row matrix.
181	Differs from `LinearAlgebra`'s [`transpose`](@ref) in that
182	the operation is not recursive.
183
184	# Examples
185	```jldoctest; setup = :(using LinearAlgebra)
186	julia> permutedims([1, 2, 3, 4])
187	1×4 Matrix{Int64}:
188	1 2 3 4
189
190	julia> V = [[[1 2; 3 4]]; [[5 6; 7 8]]]
191	2-element Vector{Matrix{Int64}}:
192	[1 2; 3 4]
193	[5 6; 7 8]
194
195	julia> permutedims(V)
196	1×2 Matrix{Matrix{Int64}}:
197	[1 2; 3 4] [5 6; 7 8]
198
199	julia> transpose(V)
200	1×2 transpose(::Vector{Matrix{Int64}}) with eltype Transpose{Int64, Matrix{Int64}}:
201	[1 3; 2 4] [5 7; 6 8]
202	```
203	"""
204	permutedims(v::AbstractVector) = reshape(v, (1, length(v)))	135✔
205
206	"""
207	permutedims!(dest, src, perm)
208
209	Permute the dimensions of array `src` and store the result in the array `dest`. `perm` is a
210	vector specifying a permutation of length `ndims(src)`. The preallocated array `dest` should
211	have `size(dest) == size(src)[perm]` and is completely overwritten. No in-place permutation
212	is supported and unexpected results will happen if `src` and `dest` have overlapping memory
213	regions.
214
215	See also [`permutedims`](@ref).
216	"""
217	function permutedims!(dest, src::AbstractArray, perm)	12✔
218	Base.checkdims_perm(dest, src, perm)	12✔
219	P = PermutedDimsArray(dest, invperm(perm))	10✔
220	_copy!(P, src)	10✔
221	return dest	10✔
222	end
223
224	function Base.copyto!(dest::PermutedDimsArray{T,N}, src::AbstractArray{T,N}) where {T,N}	×
225	checkbounds(dest, axes(src)...)	×
226	_copy!(dest, src)	×
227	end
228	Base.copyto!(dest::PermutedDimsArray, src::AbstractArray) = _copy!(dest, src)	×
229
230	function _copy!(P::PermutedDimsArray{T,N,perm}, src) where {T,N,perm}	10✔
231	# If dest/src are "close to dense," then it pays to be cache-friendly.
232	# Determine the first permuted dimension
233	d = 0 # d+1 will hold the first permuted dimension of src	10✔
234	while d < ndims(src) && perm[d+1] == d+1	16✔
235	d += 1	6✔
236	end	6✔
237	if d == ndims(src)	10✔
238	copyto!(parent(P), src) # it's not permuted	3✔
239	else
240	R1 = CartesianIndices(axes(src)[1:d])	9✔
241	d1 = findfirst(isequal(d+1), perm)::Int # first permuted dim of dest	9✔
242	R2 = CartesianIndices(axes(src)[d+2:d1-1])	8✔
243	R3 = CartesianIndices(axes(src)[d1+1:end])	8✔
244	_permutedims!(P, src, R1, R2, R3, d+1, d1)	8✔
245	end
246	return P	10✔
247	end
248
249	@noinline function _permutedims!(P::PermutedDimsArray, src, R1::CartesianIndices{0}, R2, R3, ds, dp)	7✔
250	ip, is = axes(src, dp), axes(src, ds)	7✔
251	for jo in first(ip):8:last(ip), io in first(is):8:last(is)	14✔
252	for I3 in R3, I2 in R2	11✔
253	for j in jo:min(jo+7, last(ip))	26✔
254	for i in io:min(io+7, last(is))	72✔
255	@inbounds P[i, I2, j, I3] = src[i, I2, j, I3]	108✔
256	end	180✔
257	end	59✔
258	end	17✔
259	end	7✔
260	P	7✔
261	end
262
263	@noinline function _permutedims!(P::PermutedDimsArray, src, R1, R2, R3, ds, dp)	1✔
264	ip, is = axes(src, dp), axes(src, ds)	1✔
265	for jo in first(ip):8:last(ip), io in first(is):8:last(is)	2✔
266	for I3 in R3, I2 in R2	1✔
267	for j in jo:min(jo+7, last(ip))	2✔
268	for i in io:min(io+7, last(is))	4✔
269	for I1 in R1	12✔
270	@inbounds P[I1, i, I2, j, I3] = src[I1, i, I2, j, I3]	18✔
271	end	24✔
272	end	10✔
273	end	3✔
274	end	1✔
275	end	1✔
276	P	1✔
277	end
278
279	const CommutativeOps = Union{typeof(+),typeof(Base.add_sum),typeof(min),typeof(max),typeof(Base._extrema_rf),typeof(\|),typeof(&)}
280
281	function Base._mapreduce_dim(f, op::CommutativeOps, init::Base._InitialValue, A::PermutedDimsArray, dims::Colon)	15✔
282	Base._mapreduce_dim(f, op, init, parent(A), dims)	15✔
283	end
284	function Base._mapreduce_dim(f::typeof(identity), op::Union{typeof(Base.mul_prod),typeof(*)}, init::Base._InitialValue, A::PermutedDimsArray{<:Union{Real,Complex}}, dims::Colon)	4✔
285	Base._mapreduce_dim(f, op, init, parent(A), dims)	4✔
286	end
287
288	function Base.mapreducedim!(f, op::CommutativeOps, B::AbstractArray{T,N}, A::PermutedDimsArray{S,N,perm,iperm}) where {T,S,N,perm,iperm}	10✔
289	C = PermutedDimsArray{T,N,iperm,perm,typeof(B)}(B) # make the inverse permutation for the output	10✔
290	Base.mapreducedim!(f, op, C, parent(A))	10✔
291	B	10✔
292	end
293	function Base.mapreducedim!(f::typeof(identity), op::Union{typeof(Base.mul_prod),typeof(*)}, B::AbstractArray{T,N}, A::PermutedDimsArray{<:Union{Real,Complex},N,perm,iperm}) where {T,N,perm,iperm}	8✔
294	C = PermutedDimsArray{T,N,iperm,perm,typeof(B)}(B) # make the inverse permutation for the output	8✔
295	Base.mapreducedim!(f, op, C, parent(A))	8✔
296	B	8✔
297	end
298
299	function Base.showarg(io::IO, A::PermutedDimsArray{T,N,perm}, toplevel) where {T,N,perm}	2✔
300	print(io, "PermutedDimsArray(")	2✔
301	Base.showarg(io, parent(A), false)	2✔
302	print(io, ", ", perm, ')')	2✔
303	toplevel && print(io, " with eltype ", eltype(A))	2✔
304	return nothing	2✔
305	end
306
307	end

JuliaLang / julia / #37433

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous