mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-18 21:21:17 +00:00
a061fedb5d
Comment out the affinity-setting logic which introduced an unnecessary binary size increase for the minimal build. Co-authored-by: Randy Shuai <rashuai@microsoft.com>
31 lines
No EOL
1.1 KiB
C++
31 lines
No EOL
1.1 KiB
C++
// Copyright (c) Microsoft Corporation. All rights reserved.
|
|
// Licensed under the MIT License.
|
|
|
|
#include "core/common/common.h"
|
|
#include "core/session/onnxruntime_cxx_api.h"
|
|
#include "core/session/onnxruntime_session_options_config_keys.h"
|
|
#include "core/optimizer/graph_transformer_level.h"
|
|
#include "gmock/gmock.h"
|
|
|
|
using namespace onnxruntime;
|
|
|
|
TEST(CApiTest, session_options_graph_optimization_level) {
|
|
// Test set optimization level succeeds when valid level is provided.
|
|
Ort::SessionOptions options;
|
|
options.SetGraphOptimizationLevel(ORT_ENABLE_EXTENDED);
|
|
}
|
|
|
|
#if !defined(ORT_MINIMAL_BUILD) && !defined(ORT_EXTENDED_MINIMAL_BUILD) && !defined(ORT_NO_EXCEPTIONS)
|
|
|
|
TEST(CApiTest, session_options_oversized_affinity_string) {
|
|
Ort::SessionOptions options;
|
|
std::string long_affinity_str(onnxruntime::kMaxStrLen + 1, '0');
|
|
try {
|
|
options.AddConfigEntry(kOrtSessionOptionsConfigIntraOpThreadAffinities, long_affinity_str.c_str());
|
|
ASSERT_TRUE(false) << "Creation of config should have thrown exception";
|
|
} catch (const std::exception& ex) {
|
|
ASSERT_THAT(ex.what(), testing::HasSubstr("Config value is longer than maximum length: "));
|
|
}
|
|
}
|
|
|
|
#endif |