mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-07-02 03:55:34 +00:00
8e9188e265
### Description <!-- Describe your changes. --> SessionOptions use_deterministic_compute can be set via the python API. User request to enable setting via C API. ### 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. --> #17416
37 lines
1.4 KiB
C++
37 lines
1.4 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);
|
|
}
|
|
|
|
TEST(CApiTest, session_options_deterministic_compute) {
|
|
// Manual validation currently. Check that SetDeterministicCompute in abi_session_options.cc is hit.
|
|
Ort::SessionOptions options;
|
|
options.SetDeterministicCompute(true);
|
|
}
|
|
|
|
#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
|