compile.cpp

Go to the documentation of this file.

#include "mim/plug/compile/compile.h"
 
#include <mim/config.h>
 
#include <mim/pass/beta_red.h>
#include <mim/pass/eta_exp.h>
#include <mim/pass/eta_red.h>
#include <mim/pass/lam_spec.h>
#include <mim/pass/pass.h>
#include <mim/pass/pipelinebuilder.h>
#include <mim/pass/ret_wrap.h>
#include <mim/pass/scalarize.h>
#include <mim/pass/tail_rec_elim.h>
 
#include "mim/plug/compile/pass/debug_print.h"
#include "mim/plug/compile/pass/internal_cleanup.h"
 
using namespace mim;
using namespace mim::plug;
 
void add_phases(Defs phases, World& world, Passes& passes, PipelineBuilder& builder) {
    for (auto phase : phases) compile::handle_optimization_part(phase, world, passes, builder);
}
 
void add_passes(World& world, PipelineBuilder& builder, Passes& passes, DefVec& pass_list) {
    // Concept: We create a convention that passes register in the pipeline using append_**in**_last.
    // This pass then calls the registered passes in the order they were registered in the last phase.
 
    // We create a new dummy phase in which the passes should be inserted.
    // builder.append_phase_end([](Pipeline&) {});
    builder.begin_pass_phase();
    for (auto pass : pass_list) compile::handle_optimization_part(pass, world, passes, builder);
    builder.end_pass_phase();
}
 
extern "C" MIM_EXPORT mim::Plugin mim_get_plugin() {
    return {"compile", [](Normalizers& normalizers) { compile::register_normalizers(normalizers); },
            [](Passes& passes) {
                auto debug_phase_flag = flags_t(Annex::Base<mim::plug::compile::debug_phase>);
                assert_emplace(passes, debug_phase_flag, [](World& world, PipelineBuilder& builder, const Def* app) {
                    world.DLOG("Generate debug_phase: {}", app);
                    int level = (int)(app->as<App>()->arg(0)->as<Lit>()->get<u64>());
                    world.DLOG("  Level: {}", level);
                    builder.add_phase<compile::DebugPrint>(level);
                });
 
                assert_emplace(passes, flags_t(Annex::Base<mim::plug::compile::passes_to_phase>),
                               [&](World& world, PipelineBuilder& builder, const Def* app) {
                                   auto pass_array = app->as<App>()->arg()->projs();
                                   DefVec pass_list;
                                   for (auto pass : pass_array) pass_list.push_back(pass);
                                   add_passes(world, builder, passes, pass_list);
                               });
 
                assert_emplace(passes, flags_t(Annex::Base<mim::plug::compile::phases_to_phase>),
                               [&](World& world, PipelineBuilder& builder, const Def* app) {
                                   auto phase_array = app->as<App>()->arg()->projs();
                                   DefVec phase_list;
                                   for (auto phase : phase_array) phase_list.push_back(phase);
                                   add_phases(phase_list, world, passes, builder);
                               });
 
                assert_emplace(passes, flags_t(Annex::Base<mim::plug::compile::pipe>),
                               [&](World& world, PipelineBuilder& builder, const Def* app) {
                                   auto [ax, phases] = collect_args(app);
                                   add_phases(phases, world, passes, builder);
                               });
                assert_emplace(
                    passes, flags_t(Annex::Base<mim::plug::compile::nullptr_pass>),
                    [](World&, PipelineBuilder& builder, const Def* def) { builder.def2pass(def, nullptr); });
 
                register_pass<compile::beta_red_pass, BetaRed>(passes);
                register_pass<compile::eta_red_pass, EtaRed>(passes);
 
                register_pass<compile::lam_spec_pass, LamSpec>(passes);
                register_pass<compile::ret_wrap_pass, RetWrap>(passes);
                register_pass<compile::internal_cleanup_pass, compile::InternalCleanup>(passes);
 
                register_pass_with_arg<compile::eta_exp_pass, EtaExp, EtaRed>(passes);
                register_pass_with_arg<compile::scalarize_pass, Scalarize, EtaExp>(passes);
                register_pass_with_arg<compile::tail_rec_elim_pass, TailRecElim, EtaRed>(passes);
            },
            nullptr};
}