From 9eebb05362d0f1a8e1aefc8bbc7880105395023d Mon Sep 17 00:00:00 2001 From: avik-pal Date: Sun, 17 Mar 2024 00:41:12 +0000 Subject: [PATCH] Format .jl files --- ext/BatchedRoutinesForwardDiffExt.jl | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/ext/BatchedRoutinesForwardDiffExt.jl b/ext/BatchedRoutinesForwardDiffExt.jl index 057c04a..b37405c 100644 --- a/ext/BatchedRoutinesForwardDiffExt.jl +++ b/ext/BatchedRoutinesForwardDiffExt.jl @@ -65,8 +65,7 @@ end return ForwardDiff.value.(y_duals), J_partial end -function __batched_value_and_jacobian( - ad::AutoForwardDiff, f::F, u::AbstractMatrix{T}, +function __batched_value_and_jacobian(ad::AutoForwardDiff, f::F, u::AbstractMatrix{T}, ck::Val{chunksize}) where {F, T, chunksize} N, B = size(u) @@ -116,8 +115,7 @@ end ad, f, u, Val(batched_pickchunksize(u)))) end else - jac_call = :((y, J) = __batched_value_and_jacobian( - ad, f, u, $(Val(CK)))) + jac_call = :((y, J) = __batched_value_and_jacobian(ad, f, u, $(Val(CK)))) end return Expr(:block, jac_call, :(return (y, UniformBlockDiagonalMatrix(J)))) end