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

#ifndef __CERIUM_GPU__
SchedExternTask(spinFact);
SchedExternTask(bitReverse);
SchedExternTask(norm);
SchedExternTask(butterfly);
SchedExternTask(transpose);
SchedExternTask(highPassFilter);
#endif // not __CERIUM_GPU__

void
task_init(void)
{
#ifdef __CERIUM_GPU__
    GpuSchedRegister(SPIN_FACT, "gpu/fft.cl", "spinFact");
    GpuSchedRegister(BIT_REVERSE, "gpu/fft.cl", "bitReverse");
    GpuSchedRegister(NORMALIZATION, "gpu/fft.cl", "norm");
    GpuSchedRegister(BUTTERFLY, "gpu/fft.cl", "butterfly");
    GpuSchedRegister(TRANSEPOSE, "gpu/fft.cl", "transpose");
    GpuSchedRegister(HIGH_PASS_FILTER, "gpu/fft.cl", "highPassFilter");
#else
    SchedRegisterTask(SPIN_FACT,spinFact);
    SchedRegisterTask(NORMALIZATION, norm);
    SchedRegisterTask(BIT_REVERSE, bitReverse);
    SchedRegisterTask(BUTTERFLY, butterfly);
    SchedRegisterTask(TRANSEPOSE, transpose);
    SchedRegisterTask(HIGH_PASS_FILTER, highPassFilter);
#endif
}