test metadata propagation to keyword sorter

Co-Authored-By: Shuhei Kadowaki <[email protected]>

Author: aviatesk

base/compiler/abstractinterpretation.jl

@@ -799,8 +799,7 @@ function const_prop_enabled(interp::AbstractInterpreter, sv::InferenceState, mat
         add_remark!(interp, sv, "[constprop] Disabled by parameter")
         return false
     end
-    method = match.method
-    if method.constprop == 0x02
+    if is_no_constprop(match.method)
         add_remark!(interp, sv, "[constprop] Disabled by method parameter")
         return false
     end
@@ -1003,7 +1002,7 @@ function is_all_overridden((; fargs, argtypes)::ArgInfo, sv::InferenceState)
 end
 
 function force_const_prop(interp::AbstractInterpreter, @nospecialize(f), method::Method)
-    return method.constprop == 0x01 ||
+    return is_aggressive_constprop(method) ||
            InferenceParams(interp).aggressive_constant_propagation ||
            istopfunction(f, :getproperty) ||
            istopfunction(f, :setproperty!)

base/compiler/utilities.jl

@@ -209,6 +209,20 @@ function specialize_method(match::MethodMatch; kwargs...)
     return specialize_method(match.method, match.spec_types, match.sparams; kwargs...)
 end
 
+"""
+    is_aggressive_constprop(method::Union{Method,CodeInfo}) -> Bool
+
+Check if `method` is declared as `Base.@constprop :aggressive`.
+"""
+is_aggressive_constprop(method::Union{Method,CodeInfo}) = method.constprop == 0x01
+
+"""
+    is_no_constprop(method::Union{Method,CodeInfo}) -> Bool
+
+Check if `method` is declared as `Base.@constprop :none`.
+"""
+is_no_constprop(method::Union{Method,CodeInfo}) = method.constprop == 0x02
+
 #########
 # types #
 #########

test/compiler/inline.jl

@@ -1402,3 +1402,38 @@ end
         end
     end
 end
+
+# https://github.com/JuliaLang/julia/issues/45050
+@testset "propagate :meta annotations to keyword sorter methods" begin
+    # @inline, @noinline, @constprop
+    let @inline issue45050_f1(::Any; x::Int=1) = 2x
+        @test ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(issue45050_f1)).source)
+        @test ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(Core.kwfunc(issue45050_f1))).source)
+    end
+    let @noinline issue45050_f2(::Any; x::Int=1) = 2x
+        @test !ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(issue45050_f2)).source)
+        @test !ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(Core.kwfunc(issue45050_f2))).source)
+    end
+    let Base.@constprop :aggressive issue45050_f3(::Any; x::Int=1) = 2x
+        @test Core.Compiler.is_aggressive_constprop(only(methods(issue45050_f3)))
+        @test Core.Compiler.is_aggressive_constprop(only(methods(Core.kwfunc(issue45050_f3))))
+    end
+    let Base.@constprop :none issue45050_f4(::Any; x::Int=1) = 2x
+        @test Core.Compiler.is_no_constprop(only(methods(issue45050_f4)))
+        @test Core.Compiler.is_no_constprop(only(methods(Core.kwfunc(issue45050_f4))))
+    end
+
+    # propagate multiple metadata also
+    let @inline Base.@constprop :aggressive issue45050_f5(::Any; x::Int=1) = 2x
+        @test ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(issue45050_f5)).source)
+        @test Core.Compiler.is_aggressive_constprop(only(methods(issue45050_f5)))
+        @test ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(Core.kwfunc(issue45050_f5))).source)
+        @test Core.Compiler.is_aggressive_constprop(only(methods(Core.kwfunc(issue45050_f5))))
+    end
+    let @noinline Base.@constprop :aggressive issue45050_f6(::Any; x::Int=1) = 2x
+        @test !ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(issue45050_f6)).source)
+        @test Core.Compiler.is_aggressive_constprop(only(methods(issue45050_f6)))
+        @test !ccall(:jl_ir_flag_inlineable, Bool, (Any,), only(methods(Core.kwfunc(issue45050_f6))).source)
+        @test Core.Compiler.is_aggressive_constprop(only(methods(Core.kwfunc(issue45050_f6))))
+    end
+end