Skip to content

Commit 7143f78

Browse files
committed
More name updates
1 parent c029980 commit 7143f78

File tree

3 files changed

+30
-30
lines changed

3 files changed

+30
-30
lines changed

src/cartesianproduct.jl

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -8,7 +8,7 @@ arguments(a::CartesianPair, n::Int) = arguments(a)[n]
88
arg1(a::CartesianPair) = getfield(a, :arg1)
99
arg2(a::CartesianPair) = getfield(a, :arg2)
1010

11-
×(a, b) = CartesianPair(a, b)
11+
×(a1, a2) = CartesianPair(a1, a2)
1212

1313
function Base.show(io::IO, a::CartesianPair)
1414
print(io, arg1(a), " × ", arg2(a))
@@ -37,7 +37,7 @@ function Base.show(io::IO, ::MIME"text/plain", a::CartesianProduct)
3737
return nothing
3838
end
3939

40-
×(a::AbstractVector, b::AbstractVector) = CartesianProduct(a, b)
40+
×(a1::AbstractVector, a2::AbstractVector) = CartesianProduct(a1, a2)
4141
Base.length(a::CartesianProduct) = length(arg1(a)) * length(arg2(a))
4242
Base.size(a::CartesianProduct) = (length(a),)
4343

@@ -118,12 +118,12 @@ end
118118
function CartesianProductUnitRange(p::CartesianProduct)
119119
return CartesianProductUnitRange(p, Base.OneTo(length(p)))
120120
end
121-
function CartesianProductUnitRange(a, b)
122-
return CartesianProductUnitRange(a × b)
121+
function CartesianProductUnitRange(a1, a2)
122+
return CartesianProductUnitRange(a1 × a2)
123123
end
124124
to_product_indices(a::AbstractVector) = a
125125
to_product_indices(i::Integer) = Base.OneTo(i)
126-
cartesianrange(a, b) = cartesianrange(to_product_indices(a) × to_product_indices(b))
126+
cartesianrange(a1, a2) = cartesianrange(to_product_indices(a1) × to_product_indices(a2))
127127
function cartesianrange(p::CartesianPair)
128128
p′ = to_product_indices(arg1(p)) × to_product_indices(arg2(p))
129129
return cartesianrange(p′)

src/kroneckerarray.jl

Lines changed: 21 additions & 21 deletions
Original file line numberDiff line numberDiff line change
@@ -30,14 +30,14 @@ struct KroneckerArray{T,N,A1<:AbstractArray{T,N},A2<:AbstractArray{T,N}} <:
3030
arg1::A1
3131
arg2::A2
3232
end
33-
function KroneckerArray(a::AbstractArray, b::AbstractArray)
34-
if ndims(a) != ndims(b)
33+
function KroneckerArray(a1::AbstractArray, a2::AbstractArray)
34+
if ndims(a1) != ndims(a2)
3535
throw(
3636
ArgumentError("Kronecker product requires arrays of the same number of dimensions.")
3737
)
3838
end
39-
elt = promote_type(eltype(a), eltype(b))
40-
return _convert(AbstractArray{elt}, a) _convert(AbstractArray{elt}, b)
39+
elt = promote_type(eltype(a1), eltype(a2))
40+
return _convert(AbstractArray{elt}, a1) _convert(AbstractArray{elt}, a2)
4141
end
4242
const KroneckerMatrix{T,A1<:AbstractMatrix{T},A2<:AbstractMatrix{T}} = KroneckerArray{
4343
T,2,A1,A2
@@ -204,8 +204,8 @@ function kron_nd(a::AbstractArray{<:Any,N}, b::AbstractArray{<:Any,N}) where {N}
204204
sz = reverse(ntuple(i -> size(a, i) * size(b, i), N))
205205
return permutedims(reshape(c′, sz), reverse(ntuple(identity, N)))
206206
end
207-
kron_nd(a::AbstractMatrix, b::AbstractMatrix) = kron(a, b)
208-
kron_nd(a::AbstractVector, b::AbstractVector) = kron(a, b)
207+
kron_nd(a1::AbstractMatrix, a2::AbstractMatrix) = kron(a1, a2)
208+
kron_nd(a1::AbstractVector, a2::AbstractVector) = kron(a1, a2)
209209

210210
# Eagerly collect arguments to make more general on GPU.
211211
Base.collect(a::KroneckerArray) = kron_nd(collect(arg1(a)), collect(arg2(a)))
@@ -265,10 +265,10 @@ function Base.show(io::IO, a::KroneckerArray)
265265
return nothing
266266
end
267267

268-
(a::AbstractArray, b::AbstractArray) = KroneckerArray(a, b)
269-
(a::Number, b::Number) = a * b
270-
(a::Number, b::AbstractArray) = a * b
271-
(a::AbstractArray, b::Number) = a * b
268+
(a1::AbstractArray, a2::AbstractArray) = KroneckerArray(a1, a2)
269+
(a1::Number, a2::Number) = a1 * a2
270+
(a1::Number, a2::AbstractArray) = a1 * a2
271+
(a1::AbstractArray, a2::Number) = a1 * a2
272272

273273
function Base.getindex(a::KroneckerArray, i::Integer)
274274
return a[CartesianIndices(a)[i]]
@@ -374,11 +374,11 @@ arg1(::Type{<:KroneckerStyle{<:Any,A1}}) where {A1} = A1
374374
arg1(style::KroneckerStyle) = arg1(typeof(style))
375375
arg2(::Type{<:KroneckerStyle{<:Any,<:Any,A2}}) where {A2} = A2
376376
arg2(style::KroneckerStyle) = arg2(typeof(style))
377-
function KroneckerStyle{N}(a::BroadcastStyle, b::BroadcastStyle) where {N}
378-
return KroneckerStyle{N,a,b}()
377+
function KroneckerStyle{N}(a1::BroadcastStyle, a2::BroadcastStyle) where {N}
378+
return KroneckerStyle{N,a1,a2}()
379379
end
380-
function KroneckerStyle(a::AbstractArrayStyle{N}, b::AbstractArrayStyle{N}) where {N}
381-
return KroneckerStyle{N}(a, b)
380+
function KroneckerStyle(a1::AbstractArrayStyle{N}, a2::AbstractArrayStyle{N}) where {N}
381+
return KroneckerStyle{N}(a1, a2)
382382
end
383383
function KroneckerStyle{N,A1,A2}(v::Val{M}) where {N,A1,A2,M}
384384
return KroneckerStyle{M,typeof(A1)(v),typeof(A2)(v)}()
@@ -447,11 +447,11 @@ function Broadcast.broadcasted(::KroneckerStyle, f, as...)
447447
end
448448

449449
# Linear operations.
450-
function Broadcast.broadcasted(::KroneckerStyle, ::typeof(+), a, b)
451-
return Summed(a) + Summed(b)
450+
function Broadcast.broadcasted(::KroneckerStyle, ::typeof(+), a1, a2)
451+
return Summed(a1) + Summed(a2)
452452
end
453-
function Broadcast.broadcasted(::KroneckerStyle, ::typeof(-), a, b)
454-
return Summed(a) - Summed(b)
453+
function Broadcast.broadcasted(::KroneckerStyle, ::typeof(-), a1, a2)
454+
return Summed(a1) - Summed(a2)
455455
end
456456
function Broadcast.broadcasted(::KroneckerStyle, ::typeof(*), c::Number, a)
457457
return c * Summed(a)
@@ -512,9 +512,9 @@ struct KroneckerBroadcasted{A1,A2}
512512
end
513513
@inline arg1(a::KroneckerBroadcasted) = getfield(a, :arg1)
514514
@inline arg2(a::KroneckerBroadcasted) = getfield(a, :arg2)
515-
(a::Broadcasted, b::Broadcasted) = KroneckerBroadcasted(a, b)
516-
(a::Broadcasted, b) = KroneckerBroadcasted(a, b)
517-
(a, b::Broadcasted) = KroneckerBroadcasted(a, b)
515+
(a1::Broadcasted, a2::Broadcasted) = KroneckerBroadcasted(a1, a2)
516+
(a1::Broadcasted, a2) = KroneckerBroadcasted(a1, a2)
517+
(a1, a2::Broadcasted) = KroneckerBroadcasted(a1, a2)
518518
Broadcast.materialize(a::KroneckerBroadcasted) = copy(a)
519519
Broadcast.materialize!(dest, a::KroneckerBroadcasted) = copyto!(dest, a)
520520
Broadcast.broadcastable(a::KroneckerBroadcasted) = a

src/linearalgebra.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -168,8 +168,8 @@ end
168168
Base.iterate(F::KroneckerQR) = (F.Q, Val(:R))
169169
Base.iterate(F::KroneckerQR, ::Val{:R}) = (F.R, Val(:done))
170170
Base.iterate(F::KroneckerQR, ::Val{:done}) = nothing
171-
function (a::LinearAlgebra.QRCompactWYQ, b::LinearAlgebra.QRCompactWYQ)
172-
return KroneckerQ(a, b)
171+
function (a1::LinearAlgebra.QRCompactWYQ, a2::LinearAlgebra.QRCompactWYQ)
172+
return KroneckerQ(a1, a2)
173173
end
174174
function LinearAlgebra.qr(a::KroneckerArray)
175175
Fa = qr(arg1(a))
@@ -184,8 +184,8 @@ end
184184
Base.iterate(F::KroneckerLQ) = (F.L, Val(:Q))
185185
Base.iterate(F::KroneckerLQ, ::Val{:Q}) = (F.Q, Val(:done))
186186
Base.iterate(F::KroneckerLQ, ::Val{:done}) = nothing
187-
function (a::LinearAlgebra.LQPackedQ, b::LinearAlgebra.LQPackedQ)
188-
return KroneckerQ(a, b)
187+
function (a1::LinearAlgebra.LQPackedQ, a2::LinearAlgebra.LQPackedQ)
188+
return KroneckerQ(a1, a2)
189189
end
190190
function LinearAlgebra.lq(a::KroneckerArray)
191191
Fa = lq(arg1(a))

0 commit comments

Comments
 (0)