Function eager_api_run_program

paddle/fluid/pybind/eager_custom_python_api.h:80–158

}

static PyObject *eager_api_run_program(PyObject *self,
                                       PyObject *args,
                                       PyObject *kwargs) {
  PyThreadState *tstate = nullptr;
  try {
    auto X_info = GetPyArgumentInfo("run_program", "X", args, 0, true);
    TensorListBufferAllocator X_allocator(X_info.second);
    auto &X = GetTensorListFromArgsWithBuffer("run_program",
                                              "X",
                                              0,
                                              nullptr,
                                              X_info.first,
                                              X_info.second,
                                              X_allocator);

    auto Params_info =
        GetPyArgumentInfo("run_program", "Params", args, 1, true);
    TensorListBufferAllocator Params_allocator(Params_info.second);
    auto &Params = GetTensorListFromArgsWithBuffer("run_program",
                                                   "Params",
                                                   0,
                                                   nullptr,
                                                   Params_info.first,
                                                   Params_info.second,
                                                   Params_allocator);

    auto OutScope =
        GetScopePtrListFromArgs("run_program", "OutScope", args, 2, false);
    const phi::distributed::ProcessMesh *mesh = nullptr;
    if (InputsContainDistTensor(&mesh, X, Params)) {
      X = GetTensorListFromArgsWithBuffer("run_program",
                                          "X",
                                          0,
                                          nullptr,
                                          X_info.first,
                                          X_info.second,
                                          X_allocator);
      Params = GetTensorListFromArgsWithBuffer("run_program",
                                               "Params",
                                               0,
                                               nullptr,
                                               Params_info.first,
                                               Params_info.second,
                                               Params_allocator);
    }
    VLOG(6) << "Start PIR GetProgramAttributesMapPtrFromPyArgs";
    auto prog_attrs_ptr =
        GetProgramAttributesMapPtrFromPyArgs("run_program", args, 3);
    VLOG(6) << "Finish PIR GetProgramAttributesMapPtrFromPyArgs";

    VLOG(6) << "Start PIR ConstructCudaGraphAttrMapForRunProgram";
    paddle::framework::AttributeMap cuda_graph_attrs;
    ConstructCudaGraphAttrMapForRunProgram(
        "run_program", args, 4, cuda_graph_attrs);
    VLOG(6) << "Finish PIR ConstructCudaGraphAttrMapForRunProgram";
    tstate = PyEval_SaveThread();
    auto out = egr::to_static::run_program_ad_func(
        X, Params, OutScope, *prog_attrs_ptr, cuda_graph_attrs);