remove nospecialize

JuliaDiff · Jan 9, 2024 · 6befc08 · 6befc08
1 parent 1a64879
commit 6befc08
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),
-            @nospecialize(::RuleConfig),
+            ::$RuleConfig,
             $(map(esc, primal_sig_parts)...),
         )
             return ($(esc(_with_kwargs_expr(primal_invoke, kwargs))), NoTangent())
         end
         function ChainRulesCore.frule(
-            @nospecialize(::RuleConfig), $(map(esc, primal_sig_parts)...)
+            ::$RuleConfig, $(map(esc, primal_sig_parts)...)
         )
             $(__source__)
             # Julia functions always only have 1 output, so return a single NoTangent()