flatten_stmt_seq.h

Go to the documentation of this file.

#ifndef FREE_TENSOR_FLATTEN_STMT_SEQ_H
#define FREE_TENSOR_FLATTEN_STMT_SEQ_H
 
#include <func.h>
#include <mutator.h>
 
namespace freetensor {
 
class FlattenStmtSeq : public Mutator {
    bool isEmptySeq(const Stmt &s);
 
  protected:
    Stmt visit(const StmtSeq &op) override;
    Stmt visit(const VarDef &op) override;
    Stmt visit(const For &op) override;
    Stmt visit(const If &op) override;
    Stmt visit(const Assert &op) override;
    Stmt visit(const Assume &op) override;
};
 
inline Stmt flattenStmtSeq(const Stmt &op) { return FlattenStmtSeq()(op); }
 
DEFINE_PASS_FOR_FUNC(flattenStmtSeq)
 
} // namespace freetensor
 
#endif // FREE_TENSOR_FLATTEN_STMT_SEQ_H