Add FrontIR to allowed users of Shape

src/linting/extended_checks.jl

@@ -55,7 +55,7 @@
     !is_there_any_star_marker && return x == y
 
     contains(x, "QQQ") && contains(y, "QQQ") &&
         throw(BothCannotHaveStarException("Cannot both $x and $y have a star marker"))
     if contains(x, "QQQ")
         reg_exp = Regex(replace(x, "QQQ" => ".*"))
         return !isnothing(match(reg_exp, y))
@@ -268,7 +268,7 @@
 
 function get_recommendation(msg_prefix)
     m = retrieve_full_msg_from_prefix(msg_prefix)
     m in keys(is_recommendation) || return nothing
     return is_recommendation[m]
 end
 
@@ -311,7 +311,7 @@
 end
 
 function check_for_recommendation(T::DataType, msg::String)
     @assert supertype(T) in [RecommendationExtendedRule, ViolationExtendedRule]
     b = supertype(T) == RecommendationExtendedRule
     get!(is_recommendation, msg, b)
     return nothing
@@ -343,7 +343,7 @@
 
 function check(t::NThreads_Extention, x::EXPR, markers::Dict{Symbol,String})
     # Threads.nthreads() must not be used in a const field, but it is allowed elsewhere
     haskey(markers, :const) || return
     generic_check(t, x, "Threads.nthreads()", "`Threads.nthreads()` should not be used in a constant variable.")
 end
 
@@ -397,11 +397,11 @@
 check(t::Ptr_Extension, x::EXPR) = generic_check(t, x, "Ptr{hole_variable}(hole_variable)")
 
 function check(t::ArrayWithNoType_Extension, x::EXPR, markers::Dict{Symbol,String})
     haskey(markers, :filename) || return
     contains(markers[:filename], "src/Compiler") || return
 
     haskey(markers, :macrocall) && markers[:macrocall] == "@match" && return
     haskey(markers, :macrocall) && markers[:macrocall] == "@matchrule" && return
 
     generic_check(t, x, "[]", "Need a specific Array type to be provided.")
 end
@@ -441,7 +441,7 @@
 end
 
 function check(t::Unsafe_Extension, x::EXPR, markers::Dict{Symbol,String})
     haskey(markers, :function) || return
     isnothing(match(r"_unsafe_.*", markers[:function])) || return
     isnothing(match(r"unsafe_.*", markers[:function])) || return
 
@@ -538,7 +538,7 @@
 end
 
 function check(t::RelPathAPIUsage_Extension, x::EXPR, markers::Dict{Symbol,String})
     haskey(markers, :filename) || return
     contains(markers[:filename], "src/Compiler/Front") || return
 
     generic_check(t, x, "hole_variable::RelPath", "Usage of type `RelPath` is not allowed in this context.")
@@ -550,11 +550,12 @@
 end
 
 function check(t::NonFrontShapeAPIUsage_Extension, x::EXPR, markers::Dict{Symbol,String})
     haskey(markers, :filename) || return
     # In the front-end and in FFI, we are allowed to refer to `Shape`
     contains(markers[:filename], "src/Compiler/Front") && return
     contains(markers[:filename], "src/Compiler/front2back.jl") && return
     contains(markers[:filename], "src/FFI") && return
+    contains(markers[:filename], "src/FrontIR") && return
     # Also, allow usages in tests
     contains(markers[:filename], "test/") && return
 
@@ -574,4 +575,4 @@
     msg = "Use `Threads.@threads :dynamic` instead of `Threads.@threads :static`. Static threads must not be used as generated tasks will not be able to migrate across threads."
     generic_check(t, x, "@threads :static hole_variable_star", msg)
     generic_check(t, x, "Threads.@threads :static hole_variable_star", msg)
-end
+end