Skip to content

Commit 54841b8

Browse files
authored
Fix argument naming in prepare_gradient_cache call
1 parent 58495b2 commit 54841b8

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ext/FluxMooncakeExt.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,7 @@ function Flux.gradient(f::F, adtype::AutoMooncake, args::Vararg{Any,N}) where {F
99
end
1010

1111
function Flux.withgradient(f::F, adtype::AutoMooncake, args::Vararg{Any,N}) where {F,N}
12-
cache = Mooncake.prepare_gradient_cache(f, args...; Mooncake.Config(friendly_tangents=true))
12+
cache = Mooncake.prepare_gradient_cache(f, args...; config=Mooncake.Config(friendly_tangents=true))
1313
val, grads = Mooncake.value_and_gradient!!(cache, f, args...)
1414
return (val=val, grad=grads[2:end])
1515
end

0 commit comments

Comments
 (0)