Fix AutoMooncake instantiation in test_util.jl

yebai · web-flow · commit 833dcdf4a9d7 · 2025-08-29T18:19:02.000+01:00
diff --git a/test/test_util.jl b/test/test_util.jl
@@ -56,7 +56,7 @@ function adjoint_test(
 )
     # Compute forwards-pass and j′vp.
     _f = (x) -> f(x...)
-    y, adj_ad = DI.value_and_pullback(_f, AutoMooncake(), x, ȳ)
+    y, adj_ad = DI.value_and_pullback(_f, DI.AutoMooncake(), x, ȳ)
     adj_fd = j′vp(fdm, f, ȳ, x...)
 
     # Check that forwards-pass agrees with plain forwards-pass.

Original file line number	Diff line number	Diff line change
`@@ -56,7 +56,7 @@ function adjoint_test(`
`56`	`56`	`)`
`57`	`57`	`# Compute forwards-pass and j′vp.`
`58`	`58`	`_f = (x) -> f(x...)`
`59`		`- y, adj_ad = DI.value_and_pullback(_f, AutoMooncake(), x, ȳ)`
	`59`	`+ y, adj_ad = DI.value_and_pullback(_f, DI.AutoMooncake(), x, ȳ)`
`60`	`60`	`adj_fd = j′vp(fdm, f, ȳ, x...)`
`61`	`61`
`62`	`62`	`# Check that forwards-pass agrees with plain forwards-pass.`