Method _bindCPUCore

source/backend/cpu/CPUBackend.cpp:81–114

}

void CPURuntime::_bindCPUCore() const {
    if (mCpuIds.empty()) {
        return;
    }
    auto tid = MNNGetCurrentPid();
    if (tid == mCurrentTID) {
        return;
    }
    mCurrentTID = tid;
    // Bind CPU Core
    std::vector<std::pair<const int*, int>> lockCPUIndexes(mThreadNumber);
    for (int v=0; v<mThreadNumber; ++v) {
        lockCPUIndexes[v] = std::make_pair(mCpuIds.data(), mCpuIds.size());
    }
    // Set CPU Affinity
#ifdef _OPENMP
    auto threadsNumber = mThreadNumber;
    std::vector<int> result(threadsNumber, 0);
#pragma omp parallel for
    for (int i = 0; i < threadsNumber; ++i) {
        result[i] = MNNSetSchedAffinity(lockCPUIndexes[i].first, lockCPUIndexes[i].second);
    }
#endif
#ifdef MNN_USE_THREAD_POOL
    if (nullptr != mThreadPool) {
        mThreadPool->active();
        ThreadPool::TASK task = std::make_pair([&](int i) {
            MNNSetSchedAffinity(lockCPUIndexes[i].first, lockCPUIndexes[i].second);
        }, mThreadNumber);
        mThreadPool->enqueue(&task, mTaskIndex);
        mThreadPool->deactive();
    }
#endif
}
void CPURuntime::_resetThreadPool() const {
    mThreadNumber = std::max(1, mThreadNumber);
    mThreadNumber = std::min(mThreadNumber, MAX_THREAD_NUMBER);