eta_red_phase.cpp

Go to the documentation of this file.

#include "mim/phase/eta_red_phase.h"
 
namespace mim {
 
void EtaRedPhase::rewrite_external(Def* old_mut) {
    auto new_mut = rewrite_no_eta(old_mut)->as_mut();
    if (old_mut->is_external()) new_mut->make_external();
}
 
const Def* EtaRedPhase::rewrite(const Def* old_def) {
    if (auto lam = old_def->isa<Lam>()) {
        if (auto callee = lam->eta_reduce()) {
            DLOG("eta-reduce: `{}` -> `{}`", lam, callee);
            todo_ = true;
            return rewrite(callee);
        }
    }
 
    return Rewriter::rewrite(old_def);
}
 
const Def* EtaRedPhase::rewrite_imm_Var(const Var* var) {
    return new_world().var(rewrite_no_eta(var->mut())->as_mut());
}
 
} // namespace mim