view example/bm_search/task_init.cc @ 1983:c3b4083c4467 draft

fix CudaScheduler
author Shohei KOKUBO <e105744@ie.u-ryukyu.ac.jp>
date Sat, 15 Mar 2014 17:46:13 +0900
parents a8f4227d6a21
children f786ffb2a198
line wrap: on
line source

#include "Func.h"
#include "Scheduler.h"
#ifdef __CERIUM_GPU__
#include "GpuScheduler.h"
#endif

/* 必ずこの位置に書いて */
SchedExternTask(TASK_EXEC);
SchedExternTask(READ_TASK);
SchedExternTask(BREAD_RUN_TASK_BLOCKS);
SchedExternTask(MMAP_RUN_TASK_BLOCKS);
SchedExternTask(PRINT_TASK);

/**
 * この関数は ../spe/spe-main と違って
 * 自分で呼び出せばいい関数なので
 * 好きな関数名でおk (SchedRegisterTask は必須)
 */

void
task_init(void)
{
#ifdef __CERIUM_GPU__
    GpuSchedRegister(TASK_EXEC, "gpu/Exec.cl", "run");
#endif

    SchedRegister(TASK_EXEC);
    SchedRegister(READ_TASK);
    SchedRegister(MMAP_RUN_TASK_BLOCKS);
    SchedRegister(BREAD_RUN_TASK_BLOCKS);
    SchedRegister(PRINT_TASK);
}