task-reduction-3.c 5.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218
  1. #ifdef __cplusplus
  2. extern "C"
  3. #endif
  4. void abort (void);
  5. int a[2];
  6. long long int b[7] = { 9, 11, 1, 1, 1, 13, 15 };
  7. int e[3] = { 5, 0, 5 };
  8. int f[5] = { 6, 7, 0, 0, 9 };
  9. int g[4] = { 1, 0, 0, 2 };
  10. int h[3] = { 0, 1, 4 };
  11. int k[4][2] = { { 5, 6 }, { 0, 0 }, { 0, 0 }, { 7, 8 } };
  12. long long *s;
  13. long long (*t)[2];
  14. void
  15. foo (int n, int *c, long long int *d, int m[3], int *r, int o[4], int *p, int q[4][2])
  16. {
  17. int i;
  18. for (i = 0; i < 2; i++)
  19. #pragma omp task in_reduction (+: a, c[:2]) in_reduction (*: b[2 * n:3 * n], d[0:2]) \
  20. in_reduction (+: o[n:n*2], m[1], k[1:2][:], p[0], f[2:2]) \
  21. in_reduction (+: q[1:2][:], g[n:n*2], e[1], h[0], r[2:2]) \
  22. in_reduction (*: s[1:2], t[2:2][:])
  23. {
  24. a[0] += 7;
  25. a[1] += 17;
  26. b[2] *= 2;
  27. b[4] *= 2;
  28. c[0] += 6;
  29. d[1] *= 2;
  30. e[1] += 19;
  31. f[2] += 21;
  32. f[3] += 23;
  33. g[1] += 25;
  34. g[2] += 27;
  35. h[0] += 29;
  36. k[1][0] += 31;
  37. k[2][1] += 33;
  38. m[1] += 19;
  39. r[2] += 21;
  40. r[3] += 23;
  41. o[1] += 25;
  42. o[2] += 27;
  43. p[0] += 29;
  44. q[1][0] += 31;
  45. q[2][1] += 33;
  46. s[1] *= 2;
  47. t[2][0] *= 2;
  48. t[3][1] *= 2;
  49. }
  50. }
  51. void
  52. test (int n)
  53. {
  54. int c[2] = { 0, 0 };
  55. int p[3] = { 0, 1, 4 };
  56. int q[4][2] = { { 5, 6 }, { 0, 0 }, { 0, 0 }, { 7, 8 } };
  57. long long ss[4] = { 5, 1, 1, 6 };
  58. long long tt[5][2] = { { 9, 10 }, { 11, 12 }, { 1, 1 }, { 1, 1 }, { 13, 14 } };
  59. s = ss;
  60. t = tt;
  61. #pragma omp parallel
  62. #pragma omp single
  63. {
  64. long long int d[] = { 1, 1 };
  65. int m[3] = { 5, 0, 5 };
  66. int r[5] = { 6, 7, 0, 0, 9 };
  67. int o[4] = { 1, 0, 0, 2 };
  68. #pragma omp taskgroup task_reduction (+: a, c) task_reduction (*: b[2 * n:3 * n], d) \
  69. task_reduction (+: e[1], f[2:2], g[n:n*2], h[0], k[1:2][0:2]) \
  70. task_reduction (+: o[n:n*2], m[1], q[1:2][:], p[0], r[2:2]) \
  71. task_reduction (*: t[2:2][:], s[1:n + 1])
  72. {
  73. int i;
  74. for (i = 0; i < 4; i++)
  75. #pragma omp task in_reduction (+: a, c) in_reduction (*: b[2 * n:3 * n], d) \
  76. in_reduction (+: o[n:n*2], q[1:2][:], p[0], m[1], r[2:2]) \
  77. in_reduction (+: g[n:n * 2], e[1], k[1:2][:], h[0], f[2:2]) \
  78. in_reduction (*: s[1:2], t[2:2][:])
  79. {
  80. int j;
  81. a[0] += 2;
  82. a[1] += 3;
  83. b[2] *= 2;
  84. f[3] += 8;
  85. g[1] += 9;
  86. g[2] += 10;
  87. h[0] += 11;
  88. k[1][1] += 13;
  89. k[2][1] += 15;
  90. m[1] += 16;
  91. r[2] += 8;
  92. s[1] *= 2;
  93. t[2][1] *= 2;
  94. t[3][1] *= 2;
  95. for (j = 0; j < 2; j++)
  96. #pragma omp task in_reduction (+: a, c[:2]) \
  97. in_reduction (*: b[2 * n:3 * n], d[n - 1:n + 1]) \
  98. in_reduction (+: e[1], f[2:2], g[n:n*2], h[0], k[1:2][:2]) \
  99. in_reduction (+: m[1], r[2:2], o[n:n*2], p[0], q[1:2][:2]) \
  100. in_reduction (*: s[n:2], t[2:2][:])
  101. {
  102. m[1] += 6;
  103. r[2] += 7;
  104. q[1][0] += 17;
  105. q[2][0] += 19;
  106. a[0] += 4;
  107. a[1] += 5;
  108. b[3] *= 2;
  109. b[4] *= 2;
  110. f[3] += 18;
  111. g[1] += 29;
  112. g[2] += 18;
  113. h[0] += 19;
  114. s[2] *= 2;
  115. t[2][0] *= 2;
  116. t[3][0] *= 2;
  117. foo (n, c, d, m, r, o, p, q);
  118. r[3] += 18;
  119. o[1] += 29;
  120. o[2] += 18;
  121. p[0] += 19;
  122. c[0] += 4;
  123. c[1] += 5;
  124. d[0] *= 2;
  125. e[1] += 6;
  126. f[2] += 7;
  127. k[1][0] += 17;
  128. k[2][0] += 19;
  129. }
  130. r[3] += 8;
  131. o[1] += 9;
  132. o[2] += 10;
  133. p[0] += 11;
  134. q[1][1] += 13;
  135. q[2][1] += 15;
  136. b[3] *= 2;
  137. c[0] += 4;
  138. c[1] += 9;
  139. d[0] *= 2;
  140. e[1] += 16;
  141. f[2] += 8;
  142. }
  143. }
  144. if (d[0] != 1LL << (8 + 4)
  145. || d[1] != 1LL << 16
  146. || m[0] != 5
  147. || m[1] != 19 * 16 + 6 * 8 + 16 * 4
  148. || m[2] != 5
  149. || r[0] != 6
  150. || r[1] != 7
  151. || r[2] != 21 * 16 + 7 * 8 + 8 * 4
  152. || r[3] != 23 * 16 + 18 * 8 + 8 * 4
  153. || r[4] != 9
  154. || o[0] != 1
  155. || o[1] != 25 * 16 + 29 * 8 + 9 * 4
  156. || o[2] != 27 * 16 + 18 * 8 + 10 * 4
  157. || o[3] != 2)
  158. abort ();
  159. }
  160. if (a[0] != 7 * 16 + 4 * 8 + 2 * 4
  161. || a[1] != 17 * 16 + 5 * 8 + 3 * 4
  162. || b[0] != 9 || b[1] != 11
  163. || b[2] != 1LL << (16 + 4)
  164. || b[3] != 1LL << (8 + 4)
  165. || b[4] != 1LL << (16 + 8)
  166. || b[5] != 13 || b[6] != 15
  167. || c[0] != 6 * 16 + 4 * 8 + 4 * 4
  168. || c[1] != 5 * 8 + 9 * 4
  169. || e[0] != 5
  170. || e[1] != 19 * 16 + 6 * 8 + 16 * 4
  171. || e[2] != 5
  172. || f[0] != 6
  173. || f[1] != 7
  174. || f[2] != 21 * 16 + 7 * 8 + 8 * 4
  175. || f[3] != 23 * 16 + 18 * 8 + 8 * 4
  176. || f[4] != 9
  177. || g[0] != 1
  178. || g[1] != 25 * 16 + 29 * 8 + 9 * 4
  179. || g[2] != 27 * 16 + 18 * 8 + 10 * 4
  180. || g[3] != 2
  181. || h[0] != 29 * 16 + 19 * 8 + 11 * 4
  182. || h[1] != 1 || h[2] != 4
  183. || k[0][0] != 5 || k[0][1] != 6
  184. || k[1][0] != 31 * 16 + 17 * 8
  185. || k[1][1] != 13 * 4
  186. || k[2][0] != 19 * 8
  187. || k[2][1] != 33 * 16 + 15 * 4
  188. || k[3][0] != 7 || k[3][1] != 8
  189. || p[0] != 29 * 16 + 19 * 8 + 11 * 4
  190. || p[1] != 1 || p[2] != 4
  191. || q[0][0] != 5 || q[0][1] != 6
  192. || q[1][0] != 31 * 16 + 17 * 8
  193. || q[1][1] != 13 * 4
  194. || q[2][0] != 19 * 8
  195. || q[2][1] != 33 * 16 + 15 * 4
  196. || q[3][0] != 7 || q[3][1] != 8
  197. || ss[0] != 5
  198. || ss[1] != 1LL << (16 + 4)
  199. || ss[2] != 1LL << 8
  200. || ss[3] != 6
  201. || tt[0][0] != 9 || tt[0][1] != 10 || tt[1][0] != 11 || tt[1][1] != 12
  202. || tt[2][0] != 1LL << (16 + 8)
  203. || tt[2][1] != 1LL << 4
  204. || tt[3][0] != 1LL << 8
  205. || tt[3][1] != 1LL << (16 + 4)
  206. || tt[4][0] != 13 || tt[4][1] != 14)
  207. abort ();
  208. }
  209. int
  210. main ()
  211. {
  212. test (1);
  213. return 0;
  214. }