btensor.cpp

Go to the documentation of this file.

#include "mim/plug/btensor/btensor.h"
 
#include <mim/phase.h>
#include <mim/plugin.h>
 
#include "mim/plug/btensor/phase/lower_map_reduce.h"
 
using namespace mim;
using namespace mim::plug;
 
static void reg_phases(Flags2Phases& phases) {
    Phase::hook<btensor::lower_map_reduce, btensor::phase::LowerMapReduce>(phases);
}
 
MIM_PLUGIN_ENTRY(btensor) {
    plugin.register_normalizers = btensor::register_normalizers;
    plugin.register_phases      = reg_phases;
}