Skip to content

Commit 37ba892

Browse files
committed
mark select_algorithm as non-differentiable
1 parent 07d90b3 commit 37ba892

1 file changed

Lines changed: 3 additions & 2 deletions

File tree

ext/MatrixAlgebraKitMooncakeExt/MatrixAlgebraKitMooncakeExt.jl

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -21,10 +21,11 @@ _warn_pullback_truncerror(dϵ::Real; tol = MatrixAlgebraKit.defaulttol(dϵ)) =
2121
# --------------
2222
Mooncake.tangent_type(::Type{<:AbstractAlgorithm}) = Mooncake.NoTangent
2323

24-
Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(MAK.initialize_output), Vararg{Any}}
24+
Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(MAK.select_algorithm), Any, Any, Any}
25+
Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(Core.kwcall), NamedTuple, typeof(MAK.select_algorithm), Any, Any, Any}
26+
Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(MAK.initialize_output), Any, Any, Any}
2527
Mooncake.@zero_derivative Mooncake.DefaultCtx Tuple{typeof(MAK.check_input), Vararg{Any}}
2628

27-
2829
@is_rev_primitive Tuple{typeof(copy_input), Any, Any}
2930
function rrule!!(::CoDual{typeof(copy_input)}, f_df::CoDual, A_dA::CoDual)
3031
Ac = copy_input(primal(f_df), primal(A_dA))

0 commit comments

Comments
 (0)