Skip to content

Commit 19abea2

Browse files
Update src/config.jl
1 parent dac5c53 commit 19abea2

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/config.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -175,8 +175,8 @@ This constructor does not store/modify `x`.
175175
"""
176176
function GradientConfig(f::F,
177177
x::AbstractArray{V},
178-
::Chunk{N},
179-
::T) where {F,V,N,T}
178+
::Chunk{N} = Chunk(x),
179+
::T = Tag(f, V)) where {F,V,N,T}
180180
seeds = construct_seeds(Partials{N,V})
181181
duals = similar(x, Dual{T,V,N})
182182
return GradientConfig{T,V,N,typeof(duals)}(seeds, duals)

0 commit comments

Comments
 (0)