matrix.cpp

Go to the documentation of this file.

#include "mim/plug/matrix/matrix.h"
 
#include <mim/pass.h>
#include <mim/plugin.h>
 
#include "mim/plug/matrix/pass/lower_matrix_highlevel.h"
#include "mim/plug/matrix/pass/lower_matrix_lowlevel.h"
#include "mim/plug/matrix/pass/lower_matrix_mediumlevel.h"
 
using namespace mim;
using namespace mim::plug;
 
void reg_stages(Flags2Stages& stages) {
    // clang-format off
    // phases
    Stage::hook<matrix::lower_matrix_low_level,             matrix::LowerMatrixLowLevel       >(stages);
    // passes
    Stage::hook<matrix::lower_matrix_high_level_map_reduce, matrix::LowerMatrixHighLevelMapRed>(stages);
    Stage::hook<matrix::lower_matrix_medium_level,          matrix::LowerMatrixMediumLevel    >(stages);
    // clang-format on
}
 
extern "C" MIM_EXPORT Plugin mim_get_plugin() { return {"matrix", matrix::register_normalizers, reg_stages, nullptr}; }