mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-30 23:18:20 +00:00
41dcf0d32e
### Description <!-- Describe your changes. --> 1. Add Build Info API to onnx. 2. Fix compile error while building onnxruntime_benchmark in MacOs. ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> 1. When Onnxruntime lib is serving online, we need a way to detect how this lib is built. This PR helps the developer to get the build information using `strings` such as git branch, git commit id, build type and cmake cxx flags, which is showed as follows.   If the build env has no git, there will be no git related infor:  3. Fix the following compile error while building benchmark in MacOs.  --------- Co-authored-by: Yuhong Guo <yuhong.gyh@antgroup.com>
28 lines
1.1 KiB
C++
28 lines
1.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "onnxruntime_pybind.h" // must use this for the include of <pybind11/pybind11.h>
|
|
#include <pybind11/stl.h>
|
|
#include "core/providers/get_execution_providers.h"
|
|
#include "onnxruntime_config.h"
|
|
|
|
namespace onnxruntime {
|
|
namespace python {
|
|
namespace py = pybind11;
|
|
|
|
void CreateInferencePybindStateModule(py::module& m);
|
|
|
|
PYBIND11_MODULE(onnxruntime_pybind11_state, m) {
|
|
CreateInferencePybindStateModule(m);
|
|
// move it out of shared method since training build has a little different behavior.
|
|
m.def(
|
|
"get_available_providers", []() -> const std::vector<std::string>& { return GetAvailableExecutionProviderNames(); },
|
|
"Return list of available Execution Providers in this installed version of Onnxruntime. "
|
|
"The order of elements represents the default priority order of Execution Providers "
|
|
"from highest to lowest.");
|
|
|
|
m.def("get_version_string", []() -> std::string { return ORT_VERSION; });
|
|
m.def("get_build_info", []() -> std::string { return ORT_BUILD_INFO; });
|
|
}
|
|
} // namespace python
|
|
} // namespace onnxruntime
|