chore: import upstream snapshot with attribution

2026-07-13 13:33:03 +08:00
commit 5b57521aa1
8226 changed files with 3425766 additions and 0 deletions
@@ -0,0 +1,60 @@
+#include "WorkerThread.hpp"
+#include <thread>
+
+#include <MNN/MNNDefine.h>
+//#define MNN_OPEN_TIME_TRACE
+#include <MNN/AutoTime.hpp>
+using namespace std;
+
+namespace MNN {
+WorkerThread::WorkerThread(int numberThread) {
+    for (int i=0; i<numberThread; ++i) {
+        mWorkers.emplace_back([this]() {
+            while (!mStop) {
+                Task* f = nullptr;
+                {
+                    std::unique_lock<std::mutex> _l(mQueueMutex);
+                    mCondition.wait(_l, [this] { return mStop || mTasks.size() > 0;});
+                    if (mTasks.empty()) {
+                        continue;
+                    }
+                    f = mTasks.front();
+                    mTasks.pop();
+                }
+                f->content();
+                delete f;
+            }
+        });
+    }
+}
+WorkerThread::~WorkerThread() {
+    {
+        std::lock_guard<std::mutex> _l(mQueueMutex);
+        mStop = true;
+    }
+    mCondition.notify_all();
+    for (auto& worker : mWorkers) {
+        worker.join();
+    }
+    // Complete Remain work
+    while (!mTasks.empty()) {
+        auto f = mTasks.front();
+        f->content();
+        mTasks.pop();
+        delete f;
+    }
+}
+
+bool WorkerThread::postTask(std::function<int()>&& task) {
+    {
+        AUTOTIME;
+        std::unique_lock<std::mutex> _l(mQueueMutex);
+        auto taskWrap = new Task;
+        taskWrap->content = std::move(task);
+        mTasks.push(taskWrap);
+    }
+    mCondition.notify_all();
+    return true;
+}
+
+} // namespace MNN