set filter axiom

Author: NeuralCoder3

thorin/pass/optimize.cpp

@@ -17,6 +17,7 @@ void optimize(World& world, PipelineBuilder& builder) {
     pipe.add<PassManPhase>(builder.opt_phase(world));
     pipe.add<LamSpec>();
     pipe.add<PassManPhase>(builder.codegen_prep_phase(world));
+    pipe.add<PassManPhase>(builder.opt_phase2(world));
     pipe.run();
 }
 

thorin/pass/pipelinebuilder.cpp

@@ -23,6 +23,19 @@ void PipelineBuilder::extend_codegen_prep_phase(std::function<void(PassMan&)> ex
     codegen_prep_phase_extensions_.push_back(extension);
 }
 
+std::unique_ptr<PassMan> PipelineBuilder::opt_phase2(World& world) {
+    auto man = std::make_unique<PassMan>(world);
+
+    man->add<PartialEval>();
+    man->add<BetaRed>();
+    auto er = man->add<EtaRed>();
+    auto ee = man->add<EtaExp>(er);
+    man->add<Scalerize>(ee);
+    man->add<TailRecElim>(er);
+
+    return man;
+}
+
 std::unique_ptr<PassMan> PipelineBuilder::opt_phase(World& world) {
     auto man = std::make_unique<PassMan>(world);
 

thorin/pass/pipelinebuilder.h

@@ -15,6 +15,7 @@ class PipelineBuilder {
     void extend_codegen_prep_phase(std::function<void(PassMan&)>);
 
     std::unique_ptr<PassMan> opt_phase(World& world);
+    std::unique_ptr<PassMan> opt_phase2(World& world);
     std::unique_ptr<PassMan> codegen_prep_phase(World& world);
 
 private: