Function main

tests/compile/ir/ProgramLoweringTest.cpp:34–60

};

int main() {
  mllm::initializeContext();
  {
    auto net = FooNet("foo_net");

    // Make some fake weights
    auto params = ParameterFile::create();
    for (int i = 0; i < 4; ++i) {
      auto name = "foo_net.linear_" + std::to_string(i);
      auto w = Tensor::empty({2048, 1024}).setMemType(kParamsNormal).setName(name + ".weight");
      auto b = Tensor::empty({2048}).setMemType(kParamsNormal).setName(name + ".bias");
      params->push(w.name(), w);
      params->push(b.name(), b);
    }
    net.load(params);

    auto ir_ctx = mllm::ir::trace(net, Tensor::empty({2, 1, 312, 1024}, kFloat32));

    // Lowering
    auto pm = mllm::ir::PassManager(ir_ctx);
    pm.reg(mllm::ir::createProgramLoweringPipeline());
    pm.run();

    print(ir_ctx);
  }
  mllm::memoryReport();
}