Function CompileModule

practice/cpp/5th_codegen/resnet_codegen.cc:504–551

}

void CompileModule() {
  std::cout << "\033[0;32mBegin Compile Module \033[0m" << std::endl;
  std::system("rm -rf codegen/*.so codegen/*.o;");
  std::string cmd_common;
  std::string cmd;
  std::string opt_level = " -Ofast ";
  std::string opencv_install_dir = "/usr/include/opencv4";
  std::string codegen_dir = "codegen";
  cmd_common = "g++ -mavx -fPIC " + opt_level + "-I" + opencv_install_dir +
               " -lopencv_core -lopencv_imgproc -lopencv_highgui -lopencv_imgcodecs ";

  int compile_step = 1;
  const int total_steps = 59;
  auto CompileStepVision = [&](int& step, const std::string& cmd) {
    float pre = (float)step / (float)total_steps * 100;
    auto pre_label = "[" + std::to_string((int)pre) + "%]";
    std::cout << pre_label << " \033[0;32m" << cmd << "\033[0m" << std::endl;
    step++;
  };

  // compile .cc to .o
  std::vector<std::string> obj_list;
  auto Compile2Obj = [&](const std::string& layer_name) {
    auto cmd = cmd_common + "-c " + codegen_dir + "/" + layer_name + ".cc" + " -o " + codegen_dir +
               "/" + layer_name + ".o";
    CompileStepVision(compile_step, cmd);
    std::system(cmd.c_str());
    obj_list.push_back(codegen_dir + "/" + layer_name + ".o");
  };
  Compile2Obj("maxpool");
  Compile2Obj("avgpool");
  Compile2Obj("bn");
  Compile2Obj("fc");
  Compile2Obj("codegen");
  for (int i = 0; i < 53; i++) {
    Compile2Obj("conv_" + std::to_string(i));
  }

  // link .o to libresnet.so
  cmd = cmd_common + "-shared -o " + codegen_dir + "/libresnet.so ";
  for (auto it : obj_list) {
    cmd += it + " ";
  }
  CompileStepVision(compile_step, cmd);
  std::system(cmd.c_str());
  std::system("rm -f codegen/*.o;");
  std::cout << "\033[0;32mEnd Compile Module \033[0m" << std::endl;
}

Module LoadModule() {
  void* handle = dlopen("./codegen/libresnet.so", RTLD_LAZY);