Phase.runOn is now List[ComliplationUnit] => List[ComliplationUnit]

Allows to alter number of compilation units.

1 files changed, 2 insertions, 1 deletions

diff --git a/src/dotty/tools/dotc/typer/FrontEnd.scala b/src/dotty/tools/dotc/typer/FrontEnd.scala
index 4c5e2549b..4f3b03fa1 100644
--- a/src/dotty/tools/dotc/typer/FrontEnd.scala
+++ b/src/dotty/tools/dotc/typer/FrontEnd.scala
@@ -40,13 +40,14 @@ class FrontEnd extends Phase {
     record("retainedTypedTrees", unit.tpdTree.treeSize)
   }
 
-  override def runOn(units: List[CompilationUnit])(implicit ctx: Context): Unit = {
+  override def runOn(units: List[CompilationUnit])(implicit ctx: Context): List[CompilationUnit] = {
     val unitContexts = units map (unit => ctx.fresh.setCompilationUnit(unit))
     unitContexts foreach (parse(_))
     record("parsedTrees", ast.Trees.ntrees)
     unitContexts foreach (enterSyms(_))
     unitContexts foreach (typeCheck(_))
     record("totalTrees", ast.Trees.ntrees)
+    unitContexts.map(_.compilationUnit)
   }
 
   override def run(implicit ctx: Context): Unit = {