mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-24 19:43:35 +00:00
c716b56f26
Switched the code to C++17. To build ONNX Runtime on old distros like CentOS 7, you need to install a newer GCC from additionary repos. If you build onnxruntime with the newer GCC, typically the result binary can't be distributed to other places because it depends on the new GCC's runtime libraries, something that the stock OS doesn't have. But on RHEL/CentOS, it can be better. We use Red Hat devtoolset 8/9/10 with CentOS7 building our code. The new library features(like std::filesystem) that not exists in the old C++ runtime will be statically linked into the applications with some restrictions: 1. GCC has dual ABI, but we can only use the old one. It means std::string is still copy-on-write and std::list::size() is still O(n). Also, if you build onnxruntime on CentOS 7 and link it with some binaries that were built on CentOS 8 or Ubuntu with the new ABI and export C++ symbols directly(instead of using a C API), the it won't work. 2. We still can't use std::optional. It is a limitation coming from macOS. We will solve it when we got macOS 11 build machines. It won't be too long. 3. Please avoid to use C++17 in CUDA files(*.cu). Also, the *.h files that they include(like core/framework/float16.h). This is Because CUDA 10.2 doesn't support C++17. You are welcome to use the new features in any *.cc files.
84 lines
2 KiB
C++
84 lines
2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#pragma once
|
|
|
|
#include <locale>
|
|
#include <sstream>
|
|
#include <type_traits>
|
|
|
|
#include "core/common/common.h"
|
|
|
|
namespace onnxruntime {
|
|
|
|
/**
|
|
* Tries to parse a value from an entire string.
|
|
*/
|
|
template <typename T>
|
|
bool TryParseStringWithClassicLocale(const std::string& str, T& value) {
|
|
ORT_IF_CONSTEXPR (std::is_integral<T>::value && std::is_unsigned<T>::value) {
|
|
// if T is unsigned integral type, reject negative values which will wrap
|
|
if (!str.empty() && str[0] == '-') {
|
|
return false;
|
|
}
|
|
}
|
|
|
|
// don't allow leading whitespace
|
|
if (!str.empty() && std::isspace(str[0], std::locale::classic())) {
|
|
return false;
|
|
}
|
|
|
|
std::istringstream is{str};
|
|
is.imbue(std::locale::classic());
|
|
T parsed_value{};
|
|
|
|
const bool parse_successful =
|
|
is >> parsed_value &&
|
|
is.get() == std::istringstream::traits_type::eof(); // don't allow trailing characters
|
|
if (!parse_successful) {
|
|
return false;
|
|
}
|
|
|
|
value = std::move(parsed_value);
|
|
return true;
|
|
}
|
|
|
|
inline bool TryParseStringWithClassicLocale(const std::string& str, std::string& value) {
|
|
value = str;
|
|
return true;
|
|
}
|
|
|
|
inline bool TryParseStringWithClassicLocale(const std::string& str, bool& value) {
|
|
if (str == "0" || str == "False" || str == "false") {
|
|
value = false;
|
|
return true;
|
|
}
|
|
|
|
if (str == "1" || str == "True" || str == "true") {
|
|
value = true;
|
|
return true;
|
|
}
|
|
|
|
return false;
|
|
}
|
|
|
|
/**
|
|
* Parses a value from an entire string.
|
|
*/
|
|
template <typename T>
|
|
Status ParseStringWithClassicLocale(const std::string& s, T& value) {
|
|
ORT_RETURN_IF_NOT(TryParseStringWithClassicLocale(s, value), "Failed to parse value: \"", value, "\"");
|
|
return Status::OK();
|
|
}
|
|
|
|
/**
|
|
* Parses a value from an entire string.
|
|
*/
|
|
template <typename T>
|
|
T ParseStringWithClassicLocale(const std::string& s) {
|
|
T value{};
|
|
ORT_THROW_IF_ERROR(ParseStringWithClassicLocale(s, value));
|
|
return value;
|
|
}
|
|
|
|
} // namespace onnxruntime
|