Allow Shape to be used in Front benchmarks

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))
@@ -290,7 +290,7 @@
 
 function check(t::InitializingWithFunctionRule, x::EXPR, markers::Dict{Symbol,String})
     # If we are not in a const statement, then we exit this function.
     haskey(markers, :const) || return
     generic_check(t, x, "Threads.nthreads()", "`Threads.nthreads()` should not be used in a constant variable.")
     generic_check(t, x, "is_local_deployment()", "`is_local_deployment()` should not be used in a constant variable.")
     generic_check(t, x, "Deployment.is_local_deployment()", "`Deployment.is_local_deployment()` should not be used in a constant variable.")
@@ -346,11 +346,11 @@
 check(t::PtrRule, x::EXPR) = generic_check(t, x, "Ptr{hole_variable}(hole_variable)")
 
 function check(t::ArrayWithNoTypeRule, 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
@@ -390,7 +390,7 @@
 end
 
 function check(t::UnsafeRule, x::EXPR, markers::Dict{Symbol,String})
     haskey(markers, :function) || return
     isnothing(match(r"_unsafe_.*", markers[:function])) || return
     isnothing(match(r"unsafe_.*", markers[:function])) || return
 
@@ -486,7 +486,7 @@
 end
 
 function check(t::RelPathAPIUsageRule, 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.")
@@ -498,14 +498,16 @@
 end
 
 function check(t::NonFrontShapeAPIUsageRule, 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/FrontCompiler") && return
     contains(markers[:filename], "src/FFI") && return
     contains(markers[:filename], "src/FrontIR") && return
-    # Also, allow usages in tests
+    # Allow usage in Front benchmarks
+    contains(markers[:filename], "bench/Front") && return
+    # Allow usages in tests
     contains(markers[:filename], "test/") && return
-    # Also, allow usages of the name `Shape` in `packages/` although they refer to a different thing.
+    # Allow usages of the name `Shape` in `packages/` although they refer to a different thing.
     contains(markers[:filename], "packages/RAI_Protos/src/proto/metadata.proto") && return
     contains(markers[:filename], "packages/RAI_Protos/src/gen/relationalai/protocol/metadata_pb.jl") && return
 
@@ -531,7 +533,7 @@
     is_safe_macro_call(y) =
         y.head == :macrocall && y.args[1].head == :IDENTIFIER && y.args[1].val == "@safe"
 
     is_safe_literal(x) = x.head in [:NOTHING,
                                     :INTEGER,
                                     :FLOAT,
                                     :TRUE,