Skip to content

Commit 758ac84

Browse files
format
1 parent 7983b2b commit 758ac84

23 files changed

+491
-383
lines changed

docs/pages.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,5 +3,5 @@
33
pages = [
44
"Home" => "index.md",
55
"array_types.md",
6-
"recursive_array_functions.md",
6+
"recursive_array_functions.md"
77
]

ext/RecursiveArrayToolsFastBroadcastExt.jl

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,13 +4,17 @@ using RecursiveArrayTools
44
using FastBroadcast
55
using StaticArraysCore
66

7-
const AbstractVectorOfSArray = AbstractVectorOfArray{T,N,<:AbstractVector{<:StaticArraysCore.SArray}} where {T,N}
7+
const AbstractVectorOfSArray = AbstractVectorOfArray{
8+
T, N, <:AbstractVector{<:StaticArraysCore.SArray}} where {T, N}
89

9-
@inline function FastBroadcast.fast_materialize!(::FastBroadcast.Static.False, ::DB, dst::AbstractVectorOfSArray, bc::Broadcast.Broadcasted{S}) where {S,DB}
10+
@inline function FastBroadcast.fast_materialize!(
11+
::FastBroadcast.Static.False, ::DB, dst::AbstractVectorOfSArray,
12+
bc::Broadcast.Broadcasted{S}) where {S, DB}
1013
if FastBroadcast.use_fast_broadcast(S)
1114
for i in 1:length(dst.u)
1215
unpacked = RecursiveArrayTools.unpack_voa(bc, i)
13-
dst.u[i] = StaticArraysCore.similar_type(dst.u[i])(unpacked[j] for j in eachindex(unpacked))
16+
dst.u[i] = StaticArraysCore.similar_type(dst.u[i])(unpacked[j]
17+
for j in eachindex(unpacked))
1418
end
1519
else
1620
Broadcast.materialize!(dst, bc)

ext/RecursiveArrayToolsMeasurementsExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@ import RecursiveArrayTools
44
isdefined(Base, :get_extension) ? (import Measurements) : (import ..Measurements)
55

66
function RecursiveArrayTools.recursive_unitless_bottom_eltype(a::Type{
7-
<:Measurements.Measurement,
7+
<:Measurements.Measurement,
88
})
99
typeof(oneunit(a))
1010
end

ext/RecursiveArrayToolsMonteCarloMeasurementsExt.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@ isdefined(Base, :get_extension) ? (import MonteCarloMeasurements) :
55
(import ..MonteCarloMeasurements)
66

77
function RecursiveArrayTools.recursive_unitless_bottom_eltype(a::Type{
8-
<:MonteCarloMeasurements.Particles,
8+
<:MonteCarloMeasurements.Particles,
99
})
1010
typeof(one(a))
1111
end
1212

1313
function RecursiveArrayTools.recursive_unitless_eltype(a::Type{
14-
<:MonteCarloMeasurements.Particles,
14+
<:MonteCarloMeasurements.Particles,
1515
})
1616
typeof(one(a))
1717
end

ext/RecursiveArrayToolsReverseDiffExt.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -5,13 +5,13 @@ using ReverseDiff
55
using Zygote: @adjoint
66

77
function trackedarraycopyto!(dest, src)
8-
for (i, slice) in zip(eachindex(dest.u), eachslice(src, dims=ndims(src)))
9-
if dest.u[i] isa AbstractArray
10-
dest.u[i] = reshape(reduce(vcat, slice), size(dest.u[i]))
11-
else
12-
trackedarraycopyto!(dest.u[i], slice)
13-
end
8+
for (i, slice) in zip(eachindex(dest.u), eachslice(src, dims = ndims(src)))
9+
if dest.u[i] isa AbstractArray
10+
dest.u[i] = reshape(reduce(vcat, slice), size(dest.u[i]))
11+
else
12+
trackedarraycopyto!(dest.u[i], slice)
1413
end
14+
end
1515
end
1616

1717
@adjoint function Array(VA::AbstractVectorOfArray{<:ReverseDiff.TrackedReal})

ext/RecursiveArrayToolsTrackerExt.jl

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -4,12 +4,12 @@ import RecursiveArrayTools
44
isdefined(Base, :get_extension) ? (import Tracker) : (import ..Tracker)
55

66
function RecursiveArrayTools.recursivecopy!(b::AbstractArray{T, N},
7-
a::AbstractArray{T2, N}) where {
8-
T <:
9-
Tracker.TrackedArray,
10-
T2 <:
11-
Tracker.TrackedArray,
12-
N}
7+
a::AbstractArray{T2, N}) where {
8+
T <:
9+
Tracker.TrackedArray,
10+
T2 <:
11+
Tracker.TrackedArray,
12+
N}
1313
@inbounds for i in eachindex(a)
1414
b[i] = copy(a[i])
1515
end

0 commit comments

Comments
 (0)