Test changes from FluxML/ZygoteRules.jl#26

FluxML · Oct 24, 2023 · 1d221a3 · 1d221a3
1 parent f755127
commit 1d221a3
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/Zygote.jl b/src/Zygote.jl
@@ -6,6 +6,11 @@ using LinearAlgebra: copytri!, AbstractTriangular
 import ZygoteRules: @adjoint, @adjoint!, AContext, adjoint, _pullback, pullback,
   literal_getproperty, literal_getfield, unthunk_tangent
 
+ZygoteRules.gradtuple0(::Tuple{Vararg{Nothing}}) = nothing
+ZygoteRules.gradtuple1(::Tuple{Vararg{Nothing}}) = nothing
+ZygoteRules.gradtuple2(::Tuple{Vararg{Nothing}}) = nothing
+ZygoteRules.gradtuple3(::Tuple{Vararg{Nothing}}) = nothing
+
 using ChainRulesCore
 using ChainRules: ChainRules, rrule, unthunk, canonicalize
 using IRTools