fix: set within_autodiff

avik-pal · avik-pal · commit 2a4d0f2557e5 · 2025-09-01T20:18:15.000-04:00
diff --git a/src/Enzyme.jl b/src/Enzyme.jl
@@ -314,6 +314,7 @@ function overload_autodiff(
         argprefix,
         resprefix,
         resargprefix,
+        within_autodiff=true,
     )
     (; result, linear_args, in_tys, linear_results) = mlir_fn_res
     fnwrap = mlir_fn_res.fnwrapped

Original file line number	Diff line number	Diff line change
`@@ -314,6 +314,7 @@ function overload_autodiff(`
`314`	`314`	`argprefix,`
`315`	`315`	`resprefix,`
`316`	`316`	`resargprefix,`
	`317`	`+ within_autodiff=true,`
`317`	`318`	`)`
`318`	`319`	`(; result, linear_args, in_tys, linear_results) = mlir_fn_res`
`319`	`320`	`fnwrap = mlir_fn_res.fnwrapped`