wehub-resource-sync
48 lines
1.3 KiB
C++
48 lines
1.3 KiB
C++
//
|
|
// ThreadPoolTest.cpp
|
|
// MNNTests
|
|
//
|
|
// Created by MNN on 2019/06/30.
|
|
// Copyright © 2018, Alibaba Group Holding Limited
|
|
//
|
|
|
|
#ifdef MNN_USE_THREAD_POOL
|
|
#include <MNN/MNNDefine.h>
|
|
#include "MNNTestSuite.h"
|
|
#include "backend/cpu/ThreadPool.hpp"
|
|
|
|
using namespace MNN;
|
|
|
|
class ThreadPoolTest : public MNNTestCase {
|
|
public:
|
|
virtual ~ThreadPoolTest() = default;
|
|
virtual bool run(int precision) {
|
|
std::vector<std::thread> threads;
|
|
for (int i = 0; i < 10; ++i) {
|
|
threads.emplace_back([i]() {
|
|
MNN::ThreadPool* threadPool = nullptr;
|
|
MNN::ThreadPool::init(10 - i, 0, threadPool);
|
|
// initializer
|
|
auto workIndex = threadPool->acquireWorkIndex();
|
|
FUNC_PRINT(workIndex);
|
|
threadPool->active();
|
|
ThreadPool::TASK task = std::make_pair([](int index) {
|
|
FUNC_PRINT(index);
|
|
std::this_thread::yield();
|
|
}, 10);
|
|
threadPool->enqueue(&task, workIndex);
|
|
threadPool->deactive();
|
|
threadPool->releaseWorkIndex(workIndex);
|
|
});
|
|
}
|
|
for (auto& t : threads) {
|
|
t.join();
|
|
}
|
|
MNN::ThreadPool::destroy();
|
|
return true;
|
|
}
|
|
};
|
|
|
|
MNNTestSuiteRegister(ThreadPoolTest, "core/threadpool");
|
|
#endif
|