#include "Func.h"
#include "Scheduler.h"
#include "GpuScheduler.h"

SchedExternTask(spinFact);
SchedExternTask(bitReverse);
SchedExternTask(norm);
SchedExternTask(butterfly);
SchedExternTask(transpose);
SchedExternTask(highPassFilter);

void
task_init(void)
{
    SchedRegisterTask(SPIN_FACT,spinFact);
    SchedRegisterTask(NORMALIZATION, norm);
    SchedRegisterTask(BIT_REVERSE, bitReverse);
    SchedRegisterTask(BUTTERFLY, butterfly);
    SchedRegisterTask(TRANSPOSE, transpose);
    SchedRegisterTask(HIGH_PASS_FILTER, highPassFilter);
}

void
gpu_task_init()
{
        GpuSchedRegister(SPIN_FACT, "gpu/spinFact.cl", "spinFact");
    GpuSchedRegister(BIT_REVERSE, "gpu/bitReverse.cl", "bitReverse");
    GpuSchedRegister(NORMALIZATION, "gpu/norm.cl", "norm");
    GpuSchedRegister(BUTTERFLY, "gpu/butterfly.cl", "butterfly");
    GpuSchedRegister(TRANSPOSE, "gpu/transpose.cl", "transpose");
    GpuSchedRegister(HIGH_PASS_FILTER, "gpu/highPassFilter.cl", "highPassFilter");
}