Function ApplyIrPass

paddle/fluid/framework/executor_cache.cc:171–208

}

std::unique_ptr<pir::Program> ApplyIrPass(
    pir::Program *program,
    phi::Place place,
    const std::set<std::string> &no_need_buffer_names) {
#if defined(PADDLE_WITH_CUSTOM_DEVICE)
  if (!FLAGS_enable_custom_engine.empty()) {
    std::string custom_engine_translate_pass = FLAGS_enable_custom_engine;
    std::istringstream ss(custom_engine_translate_pass);
    std::string pass;
    std::vector<std::string> passes;

    while (std::getline(ss, pass, ',')) {
      passes.push_back(pass);
      VLOG(4) << "Add CustomEngine pass : " << pass;
    }

    pir::PassManager pass_pm(pir::IrContext::Instance(), 3);
    for (std::string custom_pass : passes) {
      pass_pm.AddPass(pir::PassRegistry::Instance().Get(custom_pass));
      pass_pm.Run(program);
    }
  }
#endif
  auto ir_res = pir::PdOpLowerToKernelPass(program, place);

  if (FLAGS_pir_apply_inplace_pass) {
    pir::PassManager pm(pir::IrContext::Instance(), 3);
    pm.AddPass(pir::CreateInplacePass(no_need_buffer_names));
    pm.Run(ir_res.get());

    if (FLAGS_print_ir) {
      std::cout << "IR After inplace -------------------" << std::endl;
      std::cout << *ir_res << std::endl;
    }
  }

  return ir_res;
}

std::unique_ptr<pir::Program> ApplyRemoveShadowFeedPass(
    std::unique_ptr<pir::Program> program,