lower_matrix_highlevel.h

Go to the documentation of this file.

#pragma once
 
#include <mim/def.h>
#include <mim/pass.h>
 
namespace mim::plug::matrix {
 
/// Resolves lowering of high level operations into medium/other high-level operations.
/// Some of these transformations could be done as normalizer.
/// We rewrite matrix operations like sum, transpose, and product into `map_reduce` operations.
/// The corresponding `map_reduce` operation is looked up as `internal_mapRed_matrix_[name]`.
 
class LowerMatrixHighLevelMapRed : public RWPass<LowerMatrixHighLevelMapRed, Lam> {
public:
    LowerMatrixHighLevelMapRed(World& world, flags_t annex)
        : RWPass(world, annex) {}
 
    /// custom rewrite function
    /// memoized version of rewrite_
    const Def* rewrite(const Def*) override;
    const Def* rewrite_(const Def*);
 
private:
    Def2Def rewritten;
};
 
} // namespace mim::plug::matrix