Merge pull request #80 from RelationalAI/mm-frontcompiler-shape

RelationalAI · Sep 27, 2024 · a0e10ed · a0e10ed
2 parents 54719b4 + 28f0faf
commit a0e10ed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/src/linting/extended_checks.jl b/src/linting/extended_checks.jl
@@ -552,8 +552,7 @@ 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/FrontCompiler") && return
     contains(markers[:filename], "src/FFI") && return
     contains(markers[:filename], "src/FrontIR") && return
     # Also, allow usages in tests

diff --git a/test/rai_rules_tests.jl b/test/rai_rules_tests.jl
@@ -1926,7 +1926,7 @@ end
     """
 
     @test count_lint_errors(source; directory="/src/Execution") == 3
-    @test count_lint_errors(source; directory="/src/Compiler/Front") == 0
+    @test count_lint_errors(source; directory="/src/FrontCompiler") == 0
 
     @test lint_test(source,
         "Line 1, column 67: Usage of `Shape` is not allowed outside of the Front-end Compiler and FFI.";