Method PredictorEngine

paddle/fluid/jit/engine/predictor_engine.cc:25–55

namespace jit {

PredictorEngine::PredictorEngine(
    const std::shared_ptr<FunctionInfo> &info,
    const std::shared_ptr<VariableMap> &params_dict,
    const Place &place)
    : info_(info),
      params_dict_(params_dict),
      scope_(new framework::Scope()),
      place_(place) {
  utils::ShareParamsIntoScope(info_->ParamNames(), params_dict_, scope_.get());
  VLOG(6) << framework::GenScopeTreeDebugInfo(scope_.get());

  // TODO(Aurelius84): Expose AnalysisConfig to user.
  AnalysisConfig config;
  config.SetProgFile(info->ProgramFilePath());
  if (phi::is_gpu_place(place_)) {
    config.EnableUseGpu(100, place_.GetDeviceId());
  } else if (phi::is_cpu_place(place_)) {
    config.DisableGpu();
    config.EnableONEDNN();
    config.EnableOnednnInt8();
    config.SetOnednnCacheCapacity(0);
  }
  config.SetSkipLoadParams(true);
  config.SetApplyOptim(true);
  config.SwitchIrOptim(true);

  predictor_.reset(new AnalysisPredictor(config));

  predictor_->Init(
      scope_, std::make_shared<framework::ProgramDesc>(info_->ProgramDesc()));
}

PredictorEngine::PredictorEngine(
    const std::shared_ptr<FunctionInfo> &info,