summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authorHenning Baldersheim <balder@yahoo-inc.com>2022-12-02 19:08:27 +0100
committerGitHub <noreply@github.com>2022-12-02 19:08:27 +0100
commit6d727b4b5921a86b44c40b1306c8bb79c6f93e96 (patch)
tree9983fe83df0180f93317a34840d5a19b782dcfc1
parent3cd36dd7ac8803eee595ff1f5ddee6c1edc0a90a (diff)
parent26c99edbfd8ee2c53f9a2448e8c680e09b1cb235 (diff)
Merge pull request #25092 from vespa-engine/arnej/avoid-deprecated-onnx-method
session.GetOutputName is deprecated
-rw-r--r--eval/src/vespa/eval/onnx/onnx_wrapper.cpp29
1 files changed, 8 insertions, 21 deletions
diff --git a/eval/src/vespa/eval/onnx/onnx_wrapper.cpp b/eval/src/vespa/eval/onnx/onnx_wrapper.cpp
index ed38a385d25..7b2de4dfaa1 100644
--- a/eval/src/vespa/eval/onnx/onnx_wrapper.cpp
+++ b/eval/src/vespa/eval/onnx/onnx_wrapper.cpp
@@ -166,33 +166,20 @@ Onnx::ElementType make_element_type(ONNXTensorElementDataType element_type) {
class OnnxString {
private:
static Ort::AllocatorWithDefaultOptions _alloc;
- char *_str;
- void cleanup() {
- if (_str != nullptr) {
- _alloc.Free(_str);
- _str = nullptr;
- }
- }
- OnnxString(char *str) : _str(str) {}
+ Ort::AllocatedStringPtr _str;
+ OnnxString(Ort::AllocatedStringPtr str) : _str(std::move(str)) {}
public:
OnnxString(const OnnxString &rhs) = delete;
OnnxString &operator=(const OnnxString &rhs) = delete;
- OnnxString(OnnxString &&rhs) : _str(rhs._str) {
- rhs._str = nullptr;
- }
- OnnxString &operator=(OnnxString &&rhs) {
- cleanup();
- _str = rhs._str;
- rhs._str = nullptr;
- return *this;
- }
- const char *get() const { return _str; }
- ~OnnxString() { cleanup(); }
+ OnnxString(OnnxString &&rhs) = default;
+ OnnxString &operator=(OnnxString &&rhs) = default;
+ const char *get() const { return _str.get(); }
+ ~OnnxString() = default;
static OnnxString get_input_name(const Ort::Session &session, size_t idx) {
- return OnnxString(session.GetInputName(idx, _alloc));
+ return OnnxString(session.GetInputNameAllocated(idx, _alloc));
}
static OnnxString get_output_name(const Ort::Session &session, size_t idx) {
- return OnnxString(session.GetOutputName(idx, _alloc));
+ return OnnxString(session.GetOutputNameAllocated(idx, _alloc));
}
};
Ort::AllocatorWithDefaultOptions OnnxString::_alloc;