6
6
#include " core/session/onnxruntime_cxx_api.h"
7
7
8
8
TEST (WebAssemblyTest, test) {
9
- Ort::Env ort_env;
9
+ auto numIntraOpsThreads = 10 ;
10
+ // OrtThreadingOptions *threadingOptions;
11
+ // Ort::SessionOptions sessionOptions;
12
+ // sessionOptions.DisablePerSessionThreads();
13
+ // Ort::Env ort_env(threadingOptions, ORT_LOGGING_LEVEL_WARNING, "test");
14
+
15
+ OrtEnv* environment;
16
+ OrtThreadingOptions* envOpts = nullptr ;
17
+ Ort::GetApi ().CreateThreadingOptions (&envOpts);
18
+ Ort::GetApi ().SetGlobalIntraOpNumThreads (envOpts, numIntraOpsThreads);
19
+ Ort::GetApi ().SetGlobalInterOpNumThreads (envOpts, numInterOpsThreads);
20
+ Ort::GetApi ().SetGlobalSpinControl (envOpts, 1 );
21
+ Ort::GetApi ().CreateEnvWithGlobalThreadPools (ORT_LOGGING_LEVEL_WARNING, " test" , envOpts, &environment);
22
+ env = Ort::Env (environment);
23
+
24
+ // Ort::Env ort_env;
10
25
Ort::Session session{ort_env, " testdata/mul_1.onnx" , Ort::SessionOptions{nullptr }};
11
26
auto memory_info = Ort::MemoryInfo::CreateCpu (OrtDeviceAllocator, OrtMemTypeCPU);
12
27
@@ -39,4 +54,4 @@ TEST(WebAssemblyTest, test) {
39
54
for (size_t i = 0 ; i != total_len; ++i) {
40
55
ASSERT_EQ (expected_data[i], result[i]);
41
56
}
42
- }
57
+ }
0 commit comments