wehub-resource-sync
82 lines
2.9 KiB
C++
82 lines
2.9 KiB
C++
// Copyright (c) 2022 PaddlePaddle Authors. All Rights Reserved.
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
#pragma once
|
|
|
|
#include <iostream>
|
|
|
|
#include "paddle/fluid/eager/to_static/run_program_func.h"
|
|
#include "paddle/fluid/eager/utils.h"
|
|
#include "paddle/phi/core/enforce.h"
|
|
|
|
using egr::ConvertAllInputsToDistTensor;
|
|
using egr::InputsContainDistTensor;
|
|
|
|
namespace paddle {
|
|
namespace pybind {
|
|
|
|
static PyObject *eager_api_run_program(PyObject *self, // TOREMOVE
|
|
PyObject *args,
|
|
PyObject *kwargs) {
|
|
PyThreadState *tstate = nullptr;
|
|
try {
|
|
auto X = GetTensorListFromArgs("run_program", "X", args, 0, true);
|
|
auto Params = GetTensorListFromArgs("run_program", "Params", args, 1, true);
|
|
auto Out = GetTensorPtrListFromArgs("run_program", "Out", args, 2, true);
|
|
auto OutScope =
|
|
GetScopePtrListFromArgs("run_program", "OutScope", args, 3, false);
|
|
const phi::distributed::ProcessMesh *mesh = nullptr;
|
|
if (InputsContainDistTensor(&mesh, X, Params, Out)) {
|
|
X = GetTensorListFromArgs("run_program", "X", args, 0, true, mesh);
|
|
Params =
|
|
GetTensorListFromArgs("run_program", "Params", args, 1, true, mesh);
|
|
Out = GetTensorPtrListFromArgs("run_program", "Out", args, 2, true, mesh);
|
|
}
|
|
framework::AttributeMap attrs;
|
|
ConstructAttrMapForLegacyRunProgram(
|
|
"run_program", args, 5, PyTuple_GET_SIZE(args), attrs);
|
|
|
|
tstate = PyEval_SaveThread();
|
|
egr::to_static::legacy_run_program_ad_func(X, Params, Out, OutScope, attrs);
|
|
PyEval_RestoreThread(tstate);
|
|
tstate = nullptr;
|
|
Py_RETURN_NONE;
|
|
} catch (paddle::platform::EnforceNotMet &exception) {
|
|
if (tstate) {
|
|
PyEval_RestoreThread(tstate);
|
|
}
|
|
std::ostringstream sout;
|
|
sout << exception.what();
|
|
sout << " [operator < run_program > error]";
|
|
exception.set_error_str(sout.str());
|
|
ThrowExceptionToPython(std::current_exception());
|
|
return nullptr;
|
|
} catch (...) {
|
|
if (tstate) {
|
|
PyEval_RestoreThread(tstate);
|
|
}
|
|
ThrowExceptionToPython(std::current_exception());
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
static PyMethodDef CustomEagerMethods[] = {
|
|
{"run_program",
|
|
(PyCFunction)(void (*)(void))eager_api_run_program,
|
|
METH_VARARGS | METH_KEYWORDS,
|
|
"C++ interface function for run_program in dygraph."},
|
|
{nullptr, nullptr, 0, nullptr}};
|
|
|
|
} // namespace pybind
|
|
} // namespace paddle
|