Mercurial > hg > Members > e085722 > Cerium
diff example/dependency_task/main.cc @ 0:04e28d8d3c6f
first commit
author | Daiki KINJYO <e085722@ie.u-ryukyu.ac.jp> |
---|---|
date | Mon, 08 Nov 2010 01:23:25 +0900 |
parents | |
children |
line wrap: on
line diff
--- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/example/dependency_task/main.cc Mon Nov 08 01:23:25 2010 +0900 @@ -0,0 +1,76 @@ +#include <stdio.h> +#include <stdlib.h> +#include <string.h> +#include "TaskManager.h" +#include "Func.h" + +extern void task_init(void); + +int *idata; +int length = DATA_NUM; +int depend_flg = 1; + +const char *usr_help_str = "Usage: ./dependency [-cpu spe_num] [-nodepend]\n\ + -nodepend No Depenendcy"; + +void +run_start(TaskManager *manager) +{ + HTaskPtr t_exec[2]; + HTaskPtr t_print; + + idata = (int*)manager->allocate(sizeof(int)*length*2); + for (int i = 0; i < length*2; i++) { + idata[i] = 0; + } + + // exec1,2 が計算した配列を見る + t_print = manager->create_task(TASK_PRINT); + t_print->add_inData(idata, sizeof(int)*length*2); + t_print->add_param((memaddr)(length*2)); + + // idata を二つに分けて計算する + for (int i = 0; i < 2; i++) { + t_exec[i] = manager->create_task(TASK_EXEC); + t_exec[i]->add_inData(&idata[length*i], sizeof(int)*length); + t_exec[i]->add_outData(&idata[length*i], sizeof(int)*length); + t_exec[i]->add_param((memaddr)length); + t_exec[i]->add_param((memaddr)(13*(i+1))); + t_exec[i]->set_cpu(SPE_ANY); + + if (depend_flg) + t_print->wait_for(t_exec[i]); + } + + // add Active Queue + t_print->spawn(); + + for (int i = 0; i < 2; i++) { + t_exec[i]->spawn(); + } +} + +int +init(int argc, char **argv) +{ + for (int i = 1; argv[i]; ++i) { + if (strcmp(argv[i], "-nodepend") == 0) { + depend_flg = 0; + } + } + + return 0; +} + +int +TMmain(TaskManager *manager, int argc, char *argv[]) +{ + if (init(argc, argv) < 0) { + return -1; + } + + task_init(); + run_start(manager); + + return 0; +}