mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-24 22:17:32 +00:00
c9240f4e93
* Implement Nuphar execution provider Nuphar execution provider is a TVM-based compilation provider. It has shown great speedups for RNN models using Scan. This PR is mainly for a preview of the shared codegen library for other TVM-based providers. * Fix submodules * Fix TVM submodule * Update Nuphar to latest and resolve confliction * Remove stale files caused by merge -X theirs * Revert heap buffer change to not introduce onnxruntime_framework into onnxruntime_perf_test * Fix bad merge * Merge from Nuphar * Fix warning treated as error, revert some unnecessary changes * Revert some more test changes * Some more test revert or comments to make review easier New tests could be added later * One more revert of unnecessary changes * More change revert. Test could be added back later.
78 lines
2.2 KiB
C++
78 lines
2.2 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/codegen/common/settings.h"
|
|
|
|
#include "core/common/logging/logging.h"
|
|
#include <algorithm>
|
|
#include <cctype>
|
|
|
|
namespace onnxruntime {
|
|
namespace codegen {
|
|
|
|
CodeGenSettings& CodeGenSettings::Instance() {
|
|
static CodeGenSettings settings;
|
|
return settings;
|
|
}
|
|
|
|
CodeGenSettings::CodeGenSettings() {}
|
|
|
|
void CodeGenSettings::InsertOptions(const std::map<std::string, std::string>& options) {
|
|
for (const auto& option : options) {
|
|
const auto& key = option.first;
|
|
const auto& value = option.second;
|
|
|
|
auto iter = options_.find(key);
|
|
// found existing ones
|
|
if (iter != options_.end()) {
|
|
if (iter->second != value) {
|
|
LOGS_DEFAULT(CODEGEN_SETTINGS_LOG_LEVEL) << "CodeGenSettings: option"
|
|
<< key << " is overridded from: "
|
|
<< iter->second << " to: " << value;
|
|
iter->second = value;
|
|
}
|
|
} else {
|
|
options_.insert(std::make_pair(key, value));
|
|
}
|
|
}
|
|
}
|
|
|
|
void CodeGenSettings::DumpOptions() const {
|
|
std::ostringstream stream;
|
|
stream << "CodeGenSettings: dump all options" << std::endl;
|
|
for (const auto& option : options_) {
|
|
stream << " " << option.first << " = " << option.second << std::endl;
|
|
}
|
|
LOGS_DEFAULT(CODEGEN_SETTINGS_LOG_LEVEL) << stream.str();
|
|
}
|
|
|
|
std::string CodeGenSettings::GetOptionValue(const std::string& key) const {
|
|
const auto& iter = options_.find(key);
|
|
if (iter == options_.end()) {
|
|
LOGS_DEFAULT(CODEGEN_SETTINGS_LOG_LEVEL) << "CodeGenSettings::GetOptionValue: unrecognized option" << key;
|
|
return "";
|
|
}
|
|
return iter->second;
|
|
}
|
|
|
|
bool CodeGenSettings::HasOption(const std::string& key) const {
|
|
return options_.count(key) > 0;
|
|
}
|
|
|
|
bool CodeGenSettings::OptionMatches(const std::string& key, const std::string& value) const {
|
|
if (!HasOption(key))
|
|
return false;
|
|
|
|
#ifdef _WIN32
|
|
return 0 == _stricmp(options_.at(key).c_str(), value.c_str());
|
|
#else
|
|
return 0 == strcasecmp(options_.at(key).c_str(), value.c_str());
|
|
#endif
|
|
}
|
|
|
|
void CodeGenSettings::Clear() {
|
|
options_.clear();
|
|
}
|
|
|
|
} // namespace codegen
|
|
} // namespace onnxruntime
|