farm.h

Go to the documentation of this file.

#ifndef FARM_H
#define FARM_H

#include <iostream>
#include "task.h"
#include "src/environment.h"

namespace skepu
{


void join_farm_all()
{
    starpu_task_wait_for_all();
    getTagIds().clear();
}

void join_farm()
{
    std::vector<unsigned int> *v= &(getTagIds());
    
    std::vector<unsigned int>::iterator it = v->begin();
    for(; it!= v->end(); it++)
    {
        starpu_tag_wait(*it);
    }
    v->clear();
}


/*
* unary compose, just offload maybe, also serves as base case
*/
template<typename Function1>
void farm(Function1 *f1)
{
//  setIsFormCall(true);
    ((Task*)f1)->run_async();   //call f1
//    setIsFormCall(false);
}

#ifdef __GXX_EXPERIMENTAL_CXX0X__


unsigned count_arguments()
{
    return 0;
}

template<typename Function1, typename... FurtherFunctions>
unsigned count_arguments(Function1 *f1, FurtherFunctions... further_functions)
{
    return 1 + count_arguments(std::forward<FurtherFunctions>(further_functions)...);
}

void call_with_task()
{
}

template<typename Function1, typename... FurtherFunctions>
void call_with_task(Function1 *f1, FurtherFunctions... further_functions)
{
    ((Task*)f1)->run_async();

    //call rest
    call_with_task(std::forward<FurtherFunctions>(further_functions)...);
}


template<typename... Functions>
void farm(Functions... functions)
{
    setIsFormCall(true);
    
    call_with_task(functions...);
    
    setIsFormCall(false);
    
    if(!getUseTagId()) // join before returning control, but only if in normal mode
        starpu_task_wait_for_all();
}

#else

template<typename Function1, typename Function2>
void farm(Function1 *f1, Function2 *f2)
{
   setIsFormCall(true);
   
   ((Task*)f1)->run_async();
   ((Task*)f2)->run_async();
   
   setIsFormCall(false);
   
   if(!getUseTagId()) // join before returning control, but only if in normal mode
        starpu_task_wait_for_all();
}
#endif


}

#endif