mirror of
https://github.com/open-mmlab/mmdeploy.git
synced 2025-01-14 08:09:43 +08:00
567ec62296
* simplify deps management * skip visibility flags for MSVC * simplify cuda deps * naming * workaround for cmake<3.17 * add spdlog dependency * move the enablement of CUDA to top level CMakeLists.txt * fix MSVC build * fix lint * fix build for backend ops only * remove comment * allow to use apis/python as a standalone project * remove redundant cmake code * control shared or static lib using `MMDEPLOY_SHARED_LIBS` instead of `BUILD_SHARED_LIBS` * fix MSVC build * update docs
60 lines
1.7 KiB
C++
60 lines
1.7 KiB
C++
// Copyright (c) OpenMMLab. All rights reserved.
|
|
|
|
#include "common.h"
|
|
#include "core/utils/formatter.h"
|
|
#include "execution/execution.h"
|
|
#include "execution/schedulers/inlined_scheduler.h"
|
|
#include "execution/schedulers/registry.h"
|
|
#include "execution/schedulers/single_thread_context.h"
|
|
#include "execution/schedulers/static_thread_pool.h"
|
|
|
|
namespace mmdeploy {
|
|
|
|
namespace _python {
|
|
|
|
struct PySender {
|
|
TypeErasedSender<Value> sender_;
|
|
|
|
explicit PySender(TypeErasedSender<Value> sender) : sender_(std::move(sender)) {}
|
|
|
|
struct gil_guarded_deleter {
|
|
void operator()(py::object* p) const {
|
|
py::gil_scoped_acquire _;
|
|
delete p;
|
|
}
|
|
};
|
|
using object_ptr = std::unique_ptr<py::object, gil_guarded_deleter>;
|
|
|
|
py::object __await__() {
|
|
auto future = py::module::import("concurrent.futures").attr("Future")();
|
|
{
|
|
py::gil_scoped_release _;
|
|
StartDetached(std::move(sender_) |
|
|
Then([future = object_ptr{new py::object(future)}](const Value& value) mutable {
|
|
py::gil_scoped_acquire _;
|
|
future->attr("set_result")(ConvertToPyObject(value));
|
|
delete future.release();
|
|
}));
|
|
}
|
|
return py::module::import("asyncio").attr("wrap_future")(future).attr("__await__")();
|
|
}
|
|
};
|
|
|
|
} // namespace _python
|
|
|
|
using _python::PySender;
|
|
|
|
static void register_python_executor(py::module& m) {
|
|
py::class_<PySender, std::unique_ptr<PySender>>(m, "PySender")
|
|
.def("__await__", &PySender::__await__);
|
|
}
|
|
|
|
class PythonExecutorRegisterer {
|
|
public:
|
|
PythonExecutorRegisterer() { gPythonBindings().emplace("executor", register_python_executor); }
|
|
};
|
|
|
|
static PythonExecutorRegisterer python_executor_registerer;
|
|
|
|
} // namespace mmdeploy
|