mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-06-01 23:30:35 +00:00
a9ce941579
Description: Refine threading control options and move inter op thread pool to session state. Added thread_utils.h/cc to centralize the decision around the thread pool size under various conditions. Motivation and Context Currently the thread pool size of the parallel executor is hardcoded to 32 for some reason. This PR makes the options to configure the thread pool sizes clearer.
10 lines
No EOL
272 B
C++
10 lines
No EOL
272 B
C++
#include "core/platform/threadpool.h"
|
|
#include <memory>
|
|
#include <string>
|
|
|
|
namespace onnxruntime {
|
|
namespace concurrency {
|
|
|
|
std::unique_ptr<ThreadPool> CreateThreadPool(const std::string& name, int thread_pool_size);
|
|
} // namespace concurrency
|
|
} // namespace onnxruntime
|