Skip to content

Commit e10c6ce

Browse files
committed
format
1 parent ec1e7a5 commit e10c6ce

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

ext/PreallocationToolsForwardDiffExt.jl

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ using ArrayInterface
66
using Adapt
77

88
function PreallocationTools.dualarraycreator(u::AbstractArray{T}, siz,
9-
::Type{Val{chunk_size}}) where {T, chunk_size}
9+
::Type{Val{chunk_size}}) where {T, chunk_size}
1010
ArrayInterface.restructure(u,
1111
zeros(ForwardDiff.Dual{Nothing, T, chunk_size},
1212
siz...))
@@ -18,7 +18,8 @@ PreallocationTools.forwarddiff_compat_chunk_size(x::Int) = ForwardDiff.pickchunk
1818
PreallocationTools.chunksize(::Type{ForwardDiff.Dual{T, V, N}}) where {T, V, N} = N
1919

2020
# Define get_tmp methods for ForwardDiff.Dual types
21-
function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u::T) where {T <: ForwardDiff.Dual}
21+
function PreallocationTools.get_tmp(
22+
dc::PreallocationTools.FixedSizeDiffCache, u::T) where {T <: ForwardDiff.Dual}
2223
x = reinterpret(T, dc.dual_du)
2324
if PreallocationTools.chunksize(T) === PreallocationTools.chunksize(eltype(dc.dual_du))
2425
x
@@ -27,7 +28,8 @@ function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u
2728
end
2829
end
2930

30-
function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u::Type{T}) where {T <: ForwardDiff.Dual}
31+
function PreallocationTools.get_tmp(
32+
dc::PreallocationTools.FixedSizeDiffCache, u::Type{T}) where {T <: ForwardDiff.Dual}
3133
x = reinterpret(T, dc.dual_du)
3234
if PreallocationTools.chunksize(T) === PreallocationTools.chunksize(eltype(dc.dual_du))
3335
x
@@ -36,7 +38,8 @@ function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u
3638
end
3739
end
3840

39-
function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u::AbstractArray{T}) where {T <: ForwardDiff.Dual}
41+
function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache,
42+
u::AbstractArray{T}) where {T <: ForwardDiff.Dual}
4043
x = reinterpret(T, dc.dual_du)
4144
if PreallocationTools.chunksize(T) === PreallocationTools.chunksize(eltype(dc.dual_du))
4245
x
@@ -45,7 +48,8 @@ function PreallocationTools.get_tmp(dc::PreallocationTools.FixedSizeDiffCache, u
4548
end
4649
end
4750

48-
function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, u::T) where {T <: ForwardDiff.Dual}
51+
function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, u::T) where {T <:
52+
ForwardDiff.Dual}
4953
if isbitstype(T)
5054
nelem = div(sizeof(T), sizeof(eltype(dc.dual_du))) * length(dc.du)
5155
if nelem > length(dc.dual_du)
@@ -57,7 +61,8 @@ function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, u::T) wher
5761
end
5862
end
5963

60-
function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, ::Type{T}) where {T <: ForwardDiff.Dual}
64+
function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, ::Type{T}) where {T <:
65+
ForwardDiff.Dual}
6166
if isbitstype(T)
6267
nelem = div(sizeof(T), sizeof(eltype(dc.dual_du))) * length(dc.du)
6368
if nelem > length(dc.dual_du)
@@ -69,7 +74,8 @@ function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, ::Type{T})
6974
end
7075
end
7176

72-
function PreallocationTools.get_tmp(dc::PreallocationTools.DiffCache, u::AbstractArray{T}) where {T <: ForwardDiff.Dual}
77+
function PreallocationTools.get_tmp(
78+
dc::PreallocationTools.DiffCache, u::AbstractArray{T}) where {T <: ForwardDiff.Dual}
7379
if isbitstype(T)
7480
nelem = div(sizeof(T), sizeof(eltype(dc.dual_du))) * length(dc.du)
7581
if nelem > length(dc.dual_du)

0 commit comments

Comments
 (0)