include/skepu/src/exec_plan.h

Go to the documentation of this file.

#ifndef EXEC_PLAN_H
#define EXEC_PLAN_H

#include <map>

namespace skepu
{

enum BackEnd
{
    CPU_BACKEND,
    OMP_BACKEND,
    CU_BACKEND,
    CUM_BACKEND,
    CL_BACKEND,
    CLM_BACKEND
};


class ExecPlan
{
public:
    int maxThreads;
    int maxBlocks;
    int numOmpThreads;

    void add(int lowBound, int highBound, BackEnd backEnd)
    {
        sizePlan.insert(std::make_pair(std::make_pair(lowBound, highBound),backEnd));
    }

    BackEnd find(int size)
    {
        if(sizePlan.empty())
            return CPU_BACKEND;

        std::map< std::pair<int, int>, BackEnd >::iterator it;
        for(it = sizePlan.begin(); it != sizePlan.end(); ++it)
        {
            if(size >= it->first.first && size <= it->first.second)
                return it->second;
        }

        return (--it)->second;
    }

private:
    std::map< std::pair<int, int>, BackEnd > sizePlan;
};

}

#endif