diff --git a/src/boundary_conditions.jl b/src/boundary_conditions.jl index bb9172bd4..1480af1a3 100644 --- a/src/boundary_conditions.jl +++ b/src/boundary_conditions.jl @@ -528,6 +528,7 @@ apply_bc_temp!(bc::PressureBC, temp, β, t, setup; isright, kwargs...) = apply_bc_temp_pullback!(bc::PressureBC, φbar, β, t, setup; isright, kwargs...) = apply_bc_p_pullback!(SymmetricBC(), φbar, β, t, setup; isright, kwargs...) +# COV_EXCL_START # Wrap a function to return `nothing`, because Enzyme can not handle vector return values. function enzyme_wrap( f::Union{typeof(apply_bc_u!),typeof(apply_bc_p!),typeof(apply_bc_temp!)}, @@ -602,3 +603,4 @@ function EnzymeRules.reverse( y.dval .= x.dval # y is a copy of x return (nothing, nothing, nothing, nothing) end +# COV_EXCL_STOP diff --git a/src/operators.jl b/src/operators.jl index 1a9a1c5a8..c436d274f 100644 --- a/src/operators.jl +++ b/src/operators.jl @@ -1465,6 +1465,7 @@ function get_scale_numbers(u, setup) (; uavg, ϵ, η, λ, Reλ, L, τ) end +# COV_EXCL_START # Wrap a function to return `nothing`, because Enzyme can not handle vector return values. function enzyme_wrap( f::Union{ @@ -1753,3 +1754,4 @@ function EnzymeRules.reverse( ) @error "momentum Enzyme-AD not yet implemented" end +# COV_EXCL_STOP diff --git a/src/pressure.jl b/src/pressure.jl index 3523057a8..7405bbd8c 100644 --- a/src/pressure.jl +++ b/src/pressure.jl @@ -358,6 +358,7 @@ function psolver_spectral(setup) end end +# COV_EXCL_START # Wrap a function to return `nothing`, because Enzyme can not handle vector return values. function enzyme_wrap(f::typeof(poisson!)) function wrapped_f(p, psolve, d) @@ -393,3 +394,4 @@ function EnzymeRules.reverse( EnzymeCore.make_zero!(y.dval) return (nothing, nothing, nothing) end +# COV_EXCL_STOP diff --git a/src/sciml.jl b/src/sciml.jl index f7c1ad13b..becb909a1 100644 --- a/src/sciml.jl +++ b/src/sciml.jl @@ -46,6 +46,7 @@ function right_hand_side!(dudt, u, params_ref, t) return nothing end +# COV_EXCL_START function EnzymeRules.augmented_primal( config::RevConfigWidth{1}, func::Const{typeof(right_hand_side!)}, @@ -109,3 +110,4 @@ function EnzymeRules.reverse( return (nothing, nothing, nothing, nothing) end +# COV_EXCL_STOP