parallel_for.hpp

Go to the documentation of this file.

//
// (C) Daniel Strano and the Qrack contributors 2017, 2018. 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 General Public License V3.
// See LICENSE.md in the project root or https://www.gnu.org/licenses/gpl-3.0.en.html
// for details.

#pragma once

#include <functional>

/* Needed for bitCapInt typedefs. */
#include "qinterface.hpp"

namespace Qrack {

class ParallelFor {
private:
    int32_t numCores;

public:
    ParallelFor()
        : numCores(1)
    {
    }
    virtual ~ParallelFor() {}

    void SetConcurrencyLevel(int32_t num) { numCores = num; }
    int32_t GetConcurrencyLevel() { return numCores; }
    /*
     * Parallelization routines for spreading work across multiple cores.
     */

    typedef std::function<void(const bitCapInt, const int cpu)> ParallelFunc;
    typedef std::function<bitCapInt(const bitCapInt, const int cpu)> IncrementFunc;

    void par_for_inc(const bitCapInt begin, const bitCapInt itemCount, IncrementFunc, ParallelFunc fn);

    void par_for(const bitCapInt begin, const bitCapInt end, ParallelFunc fn);

    void par_for_skip(const bitCapInt begin, const bitCapInt end, const bitCapInt skipPower,
        const bitLenInt skipBitCount, ParallelFunc fn);

    void par_for_mask(
        const bitCapInt, const bitCapInt, const bitCapInt* maskArray, const bitLenInt maskLen, ParallelFunc fn);

    real1 par_norm(const bitCapInt maxQPower, const complex* stateArray);
};

} // namespace Qrack