tail_rec_elim.h

Go to the documentation of this file.

#pragma once
 
#include "mim/pass/pass.h"
 
namespace mim {
 
class EtaRed;
 
class TailRecElim : public FPPass<TailRecElim, Lam> {
public:
    TailRecElim(PassMan& man, EtaRed* eta_red = nullptr)
        : FPPass(man, "tail_rec_elim")
        , eta_red_(eta_red) {}
 
private:
    /// @name PassMan hooks
    ///@{
    const Def* rewrite(const Def*) override;
    undo_t analyze(const Def*) override;
    ///@}
 
    EtaRed* eta_red_;
    LamMap<std::pair<Lam*, Lam*>> old2rec_loop_;
    Lam2Lam rec2loop_;
};
 
} // namespace mim