parallel_for.hpp

Go to the documentation of this file.

//
// (C) Daniel Strano and the Qrack contributors 2017-2023. All rights reserved.
//
// This is a multithreaded, universal quantum register simulation, allowing
// (nonphysical) register cloning and direct measurement of probability and
// phase, to leverage what advantages classical emulation of qubits can have.
//
// Licensed under the GNU Lesser General Public License V3.
// See LICENSE.md in the project root or https://www.gnu.org/licenses/lgpl-3.0.en.html
// for details.
 
#pragma once
 
#include "qrack_functions.hpp"
 
namespace Qrack {
 
class ParallelFor {
private:
    const bitCapIntOcl pStride;
    bitLenInt dispatchThreshold;
    unsigned numCores;
 
public:
    ParallelFor();
 
    void SetConcurrencyLevel(unsigned num)
    {
        if (numCores == num) {
            return;
        }
        numCores = num;
        const bitLenInt pStridePow = log2(pStride);
        const bitLenInt minStridePow = (numCores > 1U) ? (bitLenInt)pow2Ocl(log2(numCores - 1U)) : 0U;
        dispatchThreshold = (pStridePow > minStridePow) ? (pStridePow - minStridePow) : 0U;
    }
    unsigned GetConcurrencyLevel() { return numCores; }
    bitCapIntOcl GetStride() { return pStride; }
    bitLenInt GetPreferredConcurrencyPower() { return dispatchThreshold; }
    /*
     * Parallelization routines for spreading work across multiple cores.
     */
 
    void par_for_inc(const bitCapIntOcl begin, const bitCapIntOcl itemCount, IncrementFunc, ParallelFunc fn);
 
    void par_for(const bitCapIntOcl begin, const bitCapIntOcl end, ParallelFunc fn);
 
    void par_for_skip(const bitCapIntOcl begin, const bitCapIntOcl end, const bitCapIntOcl skipPower,
        const bitLenInt skipBitCount, ParallelFunc fn);
 
    void par_for_mask(
        const bitCapIntOcl, const bitCapIntOcl, const std::vector<bitCapIntOcl>& maskArray, ParallelFunc fn);
 
    void par_for_set(const std::set<bitCapIntOcl>& sparseSet, ParallelFunc fn);
 
    void par_for_set(const std::vector<bitCapIntOcl>& sparseSet, ParallelFunc fn);
 
    void par_for_sparse_compose(const std::vector<bitCapIntOcl>& lowSet, const std::vector<bitCapIntOcl>& highSet,
        const bitLenInt& highStart, ParallelFunc fn);
 
    real1_f par_norm(const bitCapIntOcl maxQPower, const StateVectorPtr stateArray, real1_f norm_thresh = ZERO_R1_F);
 
    real1_f par_norm_exact(const bitCapIntOcl maxQPower, const StateVectorPtr stateArray);
};
 
} // namespace Qrack