|
1242
|
1 #include <stdio.h>
|
|
|
2 #include <stdlib.h>
|
|
|
3 #include <string.h>
|
|
|
4 #include "TaskManager.h"
|
|
|
5 #include "Func.h"
|
|
|
6
|
|
|
7 typedef unsigned long long U64;
|
|
|
8
|
|
|
9 /* task_initを宣言 */
|
|
|
10 extern void task_init(void);
|
|
|
11
|
|
|
12 /* TaskManagerを宣言 */
|
|
|
13 extern TaskManager *manager;
|
|
|
14
|
|
|
15 static U64 num = 1000;
|
|
|
16 static bool print_flag = false;
|
|
|
17
|
|
|
18 /* help文章 */
|
|
|
19 const char *usr_help_str = "Usage: ./prime [-cpu spe_num] [-num N]\n\
|
|
|
20 -cpu Number of SPE (default 1) \n\
|
|
|
21 -num Caluculate of Prime scope (default 1000 * 1000 * 1000) ";
|
|
|
22
|
|
|
23 int
|
|
|
24 init(int argc, char **argv)
|
|
|
25 {
|
|
|
26 for (int i = 1; argv[i]; ++i) {
|
|
|
27 if (strcmp(argv[i], "-num") == 0) {
|
|
|
28 num = atoi(argv[++i]);
|
|
|
29 }
|
|
|
30 else if (strcmp(argv[i], "-print") == 0) {
|
|
|
31 print_flag = true;
|
|
|
32 }
|
|
|
33 }
|
|
|
34 return 0;
|
|
|
35 }
|
|
|
36
|
|
|
37 void
|
|
|
38 prime_init(TaskManager *manager)
|
|
|
39 {
|
|
|
40
|
|
|
41 U64 div_size = 1000;
|
|
|
42 U64 task_num = ((num >> 1) + div_size - 1) / div_size;
|
|
|
43
|
|
|
44 bool *output = (bool*)manager->allocate(sizeof(bool)*task_num*div_size); /* 判定結果を収める配列 */
|
|
|
45
|
|
|
46 HTask *print = manager->create_task(PrintTask);
|
|
|
47
|
|
|
48 for (U64 i = 0; i < task_num; i++) {
|
|
|
49
|
|
|
50 HTask *prime = manager->create_task(Prime);
|
|
|
51
|
|
|
52 prime->set_outData(0,&output[i*div_size],sizeof(bool)*div_size);
|
|
|
53
|
|
|
54 prime->set_cpu(SPE_ANY);
|
|
|
55
|
|
|
56 prime->set_param(0,(memaddr)(i*div_size)); // 開始地点
|
|
|
57 prime->set_param(1,(memaddr)((i+1)*div_size - 1)); // 終了地点
|
|
|
58
|
|
|
59 print->wait_for(prime);
|
|
|
60
|
|
|
61 prime->spawn();
|
|
|
62 }
|
|
|
63
|
|
|
64 /* 出力用のタスクに判定結果を収めた配列を渡す */
|
|
|
65 print->set_inData(0,output,sizeof(bool)*task_num*div_size);
|
|
|
66 /* 出力する数を渡す */
|
|
|
67 print->set_param(0,(memaddr)num);
|
|
|
68 /* printするかどうかを渡す */
|
|
|
69 print->set_param(1,(memaddr)print_flag);
|
|
|
70 /* PPEを使うように指示 */
|
|
|
71 print->set_cpu(CPU_PPE);
|
|
|
72 /* タスクを登録 */
|
|
|
73 print->spawn();
|
|
|
74 }
|
|
|
75
|
|
|
76 int
|
|
|
77 TMmain(TaskManager *manager, int argc, char *argv[])
|
|
|
78 {
|
|
|
79 if (init(argc, argv) < 0) {
|
|
|
80 return -1;
|
|
|
81 }
|
|
|
82
|
|
|
83 task_init();
|
|
|
84 prime_init(manager);
|
|
|
85
|
|
|
86 return 0;
|
|
|
87 }
|
