nospecialize on tuple

JuliaDiff · Jan 9, 2024 · 161e5f1 · 161e5f1
1 parent 67036a8
commit 161e5f1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/rule_definition_tools.jl b/src/rule_definition_tools.jl
@@ -403,13 +403,13 @@ function _nondiff_frule_expr(__source__, primal_sig_parts, primal_invoke)
         function (::Core.kwftype(typeof(ChainRulesCore.frule)))(
             @nospecialize($kwargs::Any),
             frule::typeof(ChainRulesCore.frule),
-            ::Tuple,
+            @nospecialize(::Tuple),
             $(map(esc, primal_sig_parts)...),
         )
             return ($(esc(_with_kwargs_expr(primal_invoke, kwargs))), NoTangent())
         end
         function ChainRulesCore.frule(
-            ::Tuple, $(map(esc, primal_sig_parts)...)
+            @nospecialize(::Tuple), $(map(esc, primal_sig_parts)...)
         )
             $(__source__)
             # Julia functions always only have 1 output, so return a single NoTangent()