Skip to content

Commit 4338da2

Browse files
author
Paco Wong
committed
Create thread pool
1 parent daefbd4 commit 4338da2

File tree

1 file changed

+17
-2
lines changed

1 file changed

+17
-2
lines changed

onnxruntime/test/wasm/test_inference.cc

+17-2
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,22 @@
66
#include "core/session/onnxruntime_cxx_api.h"
77

88
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;
1025
Ort::Session session{ort_env, "testdata/mul_1.onnx", Ort::SessionOptions{nullptr}};
1126
auto memory_info = Ort::MemoryInfo::CreateCpu(OrtDeviceAllocator, OrtMemTypeCPU);
1227

@@ -39,4 +54,4 @@ TEST(WebAssemblyTest, test) {
3954
for (size_t i = 0; i != total_len; ++i) {
4055
ASSERT_EQ(expected_data[i], result[i]);
4156
}
42-
}
57+
}

0 commit comments

Comments
 (0)