Skip to content

Commit bc48682

Browse files
committed
Add config
1 parent e5edb0b commit bc48682

2 files changed

Lines changed: 9 additions & 4 deletions

File tree

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/forward_onearg.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,9 @@ function DI.prepare_pushforward_nokwarg(
1616
) where {F,C}
1717
_sig = DI.signature(f, backend, x, tx, contexts...; strict)
1818
config = get_config(backend)
19-
# TODO: silence_debug_messages
20-
cache = prepare_derivative_cache(f, x, map(DI.unwrap, contexts)...; config.debug_mode)
19+
cache = prepare_derivative_cache(
20+
f, x, map(DI.unwrap, contexts)...; config.debug_mode, config.silence_debug_messages
21+
)
2122
dx_righttype = zero_tangent(x)
2223
prep = MooncakeOneArgPushforwardPrep(_sig, cache, dx_righttype)
2324
return prep

DifferentiationInterface/ext/DifferentiationInterfaceMooncakeExt/forward_twoarg.jl

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -18,9 +18,13 @@ function DI.prepare_pushforward_nokwarg(
1818
) where {F,C}
1919
_sig = DI.signature(f!, y, backend, x, tx, contexts...; strict)
2020
config = get_config(backend)
21-
# TODO: silence_debug_messages
2221
cache = prepare_derivative_cache(
23-
f!, y, x, map(DI.unwrap, contexts)...; config.debug_mode
22+
f!,
23+
y,
24+
x,
25+
map(DI.unwrap, contexts)...;
26+
config.debug_mode,
27+
config.silence_debug_messages,
2428
)
2529
dx_righttype = zero_tangent(x)
2630
dy_righttype = zero_tangent(y)

0 commit comments

Comments
 (0)