annotate libgfortran/generated/sum_i2.c @ 158:494b0b89df80 default tip

...
author Shinji KONO <kono@ie.u-ryukyu.ac.jp>
date Mon, 25 May 2020 18:13:55 +0900
parents 1830386684a0
children
Ignore whitespace changes - Everywhere: Within whitespace: At end of lines:
rev   line source
111
kono
parents:
diff changeset
1 /* Implementation of the SUM intrinsic
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
2 Copyright (C) 2002-2020 Free Software Foundation, Inc.
111
kono
parents:
diff changeset
3 Contributed by Paul Brook <paul@nowt.org>
kono
parents:
diff changeset
4
kono
parents:
diff changeset
5 This file is part of the GNU Fortran 95 runtime library (libgfortran).
kono
parents:
diff changeset
6
kono
parents:
diff changeset
7 Libgfortran is free software; you can redistribute it and/or
kono
parents:
diff changeset
8 modify it under the terms of the GNU General Public
kono
parents:
diff changeset
9 License as published by the Free Software Foundation; either
kono
parents:
diff changeset
10 version 3 of the License, or (at your option) any later version.
kono
parents:
diff changeset
11
kono
parents:
diff changeset
12 Libgfortran is distributed in the hope that it will be useful,
kono
parents:
diff changeset
13 but WITHOUT ANY WARRANTY; without even the implied warranty of
kono
parents:
diff changeset
14 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
kono
parents:
diff changeset
15 GNU General Public License for more details.
kono
parents:
diff changeset
16
kono
parents:
diff changeset
17 Under Section 7 of GPL version 3, you are granted additional
kono
parents:
diff changeset
18 permissions described in the GCC Runtime Library Exception, version
kono
parents:
diff changeset
19 3.1, as published by the Free Software Foundation.
kono
parents:
diff changeset
20
kono
parents:
diff changeset
21 You should have received a copy of the GNU General Public License and
kono
parents:
diff changeset
22 a copy of the GCC Runtime Library Exception along with this program;
kono
parents:
diff changeset
23 see the files COPYING3 and COPYING.RUNTIME respectively. If not, see
kono
parents:
diff changeset
24 <http://www.gnu.org/licenses/>. */
kono
parents:
diff changeset
25
kono
parents:
diff changeset
26 #include "libgfortran.h"
kono
parents:
diff changeset
27
kono
parents:
diff changeset
28
kono
parents:
diff changeset
29 #if defined (HAVE_GFC_INTEGER_2) && defined (HAVE_GFC_INTEGER_2)
kono
parents:
diff changeset
30
kono
parents:
diff changeset
31
kono
parents:
diff changeset
32 extern void sum_i2 (gfc_array_i2 * const restrict,
kono
parents:
diff changeset
33 gfc_array_i2 * const restrict, const index_type * const restrict);
kono
parents:
diff changeset
34 export_proto(sum_i2);
kono
parents:
diff changeset
35
kono
parents:
diff changeset
36 void
kono
parents:
diff changeset
37 sum_i2 (gfc_array_i2 * const restrict retarray,
kono
parents:
diff changeset
38 gfc_array_i2 * const restrict array,
kono
parents:
diff changeset
39 const index_type * const restrict pdim)
kono
parents:
diff changeset
40 {
kono
parents:
diff changeset
41 index_type count[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
42 index_type extent[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
43 index_type sstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
44 index_type dstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
45 const GFC_INTEGER_2 * restrict base;
kono
parents:
diff changeset
46 GFC_INTEGER_2 * restrict dest;
kono
parents:
diff changeset
47 index_type rank;
kono
parents:
diff changeset
48 index_type n;
kono
parents:
diff changeset
49 index_type len;
kono
parents:
diff changeset
50 index_type delta;
kono
parents:
diff changeset
51 index_type dim;
kono
parents:
diff changeset
52 int continue_loop;
kono
parents:
diff changeset
53
kono
parents:
diff changeset
54 /* Make dim zero based to avoid confusion. */
kono
parents:
diff changeset
55 rank = GFC_DESCRIPTOR_RANK (array) - 1;
kono
parents:
diff changeset
56 dim = (*pdim) - 1;
kono
parents:
diff changeset
57
kono
parents:
diff changeset
58 if (unlikely (dim < 0 || dim > rank))
kono
parents:
diff changeset
59 {
kono
parents:
diff changeset
60 runtime_error ("Dim argument incorrect in SUM intrinsic: "
kono
parents:
diff changeset
61 "is %ld, should be between 1 and %ld",
kono
parents:
diff changeset
62 (long int) dim + 1, (long int) rank + 1);
kono
parents:
diff changeset
63 }
kono
parents:
diff changeset
64
kono
parents:
diff changeset
65 len = GFC_DESCRIPTOR_EXTENT(array,dim);
kono
parents:
diff changeset
66 if (len < 0)
kono
parents:
diff changeset
67 len = 0;
kono
parents:
diff changeset
68 delta = GFC_DESCRIPTOR_STRIDE(array,dim);
kono
parents:
diff changeset
69
kono
parents:
diff changeset
70 for (n = 0; n < dim; n++)
kono
parents:
diff changeset
71 {
kono
parents:
diff changeset
72 sstride[n] = GFC_DESCRIPTOR_STRIDE(array,n);
kono
parents:
diff changeset
73 extent[n] = GFC_DESCRIPTOR_EXTENT(array,n);
kono
parents:
diff changeset
74
kono
parents:
diff changeset
75 if (extent[n] < 0)
kono
parents:
diff changeset
76 extent[n] = 0;
kono
parents:
diff changeset
77 }
kono
parents:
diff changeset
78 for (n = dim; n < rank; n++)
kono
parents:
diff changeset
79 {
kono
parents:
diff changeset
80 sstride[n] = GFC_DESCRIPTOR_STRIDE(array, n + 1);
kono
parents:
diff changeset
81 extent[n] = GFC_DESCRIPTOR_EXTENT(array, n + 1);
kono
parents:
diff changeset
82
kono
parents:
diff changeset
83 if (extent[n] < 0)
kono
parents:
diff changeset
84 extent[n] = 0;
kono
parents:
diff changeset
85 }
kono
parents:
diff changeset
86
kono
parents:
diff changeset
87 if (retarray->base_addr == NULL)
kono
parents:
diff changeset
88 {
kono
parents:
diff changeset
89 size_t alloc_size, str;
kono
parents:
diff changeset
90
kono
parents:
diff changeset
91 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
92 {
kono
parents:
diff changeset
93 if (n == 0)
kono
parents:
diff changeset
94 str = 1;
kono
parents:
diff changeset
95 else
kono
parents:
diff changeset
96 str = GFC_DESCRIPTOR_STRIDE(retarray,n-1) * extent[n-1];
kono
parents:
diff changeset
97
kono
parents:
diff changeset
98 GFC_DIMENSION_SET(retarray->dim[n], 0, extent[n] - 1, str);
kono
parents:
diff changeset
99
kono
parents:
diff changeset
100 }
kono
parents:
diff changeset
101
kono
parents:
diff changeset
102 retarray->offset = 0;
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
103 retarray->dtype.rank = rank;
111
kono
parents:
diff changeset
104
kono
parents:
diff changeset
105 alloc_size = GFC_DESCRIPTOR_STRIDE(retarray,rank-1) * extent[rank-1];
kono
parents:
diff changeset
106
kono
parents:
diff changeset
107 retarray->base_addr = xmallocarray (alloc_size, sizeof (GFC_INTEGER_2));
kono
parents:
diff changeset
108 if (alloc_size == 0)
kono
parents:
diff changeset
109 {
kono
parents:
diff changeset
110 /* Make sure we have a zero-sized array. */
kono
parents:
diff changeset
111 GFC_DIMENSION_SET(retarray->dim[0], 0, -1, 1);
kono
parents:
diff changeset
112 return;
kono
parents:
diff changeset
113
kono
parents:
diff changeset
114 }
kono
parents:
diff changeset
115 }
kono
parents:
diff changeset
116 else
kono
parents:
diff changeset
117 {
kono
parents:
diff changeset
118 if (rank != GFC_DESCRIPTOR_RANK (retarray))
kono
parents:
diff changeset
119 runtime_error ("rank of return array incorrect in"
kono
parents:
diff changeset
120 " SUM intrinsic: is %ld, should be %ld",
kono
parents:
diff changeset
121 (long int) (GFC_DESCRIPTOR_RANK (retarray)),
kono
parents:
diff changeset
122 (long int) rank);
kono
parents:
diff changeset
123
kono
parents:
diff changeset
124 if (unlikely (compile_options.bounds_check))
kono
parents:
diff changeset
125 bounds_ifunction_return ((array_t *) retarray, extent,
kono
parents:
diff changeset
126 "return value", "SUM");
kono
parents:
diff changeset
127 }
kono
parents:
diff changeset
128
kono
parents:
diff changeset
129 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
130 {
kono
parents:
diff changeset
131 count[n] = 0;
kono
parents:
diff changeset
132 dstride[n] = GFC_DESCRIPTOR_STRIDE(retarray,n);
kono
parents:
diff changeset
133 if (extent[n] <= 0)
kono
parents:
diff changeset
134 return;
kono
parents:
diff changeset
135 }
kono
parents:
diff changeset
136
kono
parents:
diff changeset
137 base = array->base_addr;
kono
parents:
diff changeset
138 dest = retarray->base_addr;
kono
parents:
diff changeset
139
kono
parents:
diff changeset
140 continue_loop = 1;
kono
parents:
diff changeset
141 while (continue_loop)
kono
parents:
diff changeset
142 {
kono
parents:
diff changeset
143 const GFC_INTEGER_2 * restrict src;
kono
parents:
diff changeset
144 GFC_INTEGER_2 result;
kono
parents:
diff changeset
145 src = base;
kono
parents:
diff changeset
146 {
kono
parents:
diff changeset
147
kono
parents:
diff changeset
148 result = 0;
kono
parents:
diff changeset
149 if (len <= 0)
kono
parents:
diff changeset
150 *dest = 0;
kono
parents:
diff changeset
151 else
kono
parents:
diff changeset
152 {
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
153 #if ! defined HAVE_BACK_ARG
111
kono
parents:
diff changeset
154 for (n = 0; n < len; n++, src += delta)
kono
parents:
diff changeset
155 {
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
156 #endif
111
kono
parents:
diff changeset
157
kono
parents:
diff changeset
158 result += *src;
kono
parents:
diff changeset
159 }
kono
parents:
diff changeset
160
kono
parents:
diff changeset
161 *dest = result;
kono
parents:
diff changeset
162 }
kono
parents:
diff changeset
163 }
kono
parents:
diff changeset
164 /* Advance to the next element. */
kono
parents:
diff changeset
165 count[0]++;
kono
parents:
diff changeset
166 base += sstride[0];
kono
parents:
diff changeset
167 dest += dstride[0];
kono
parents:
diff changeset
168 n = 0;
kono
parents:
diff changeset
169 while (count[n] == extent[n])
kono
parents:
diff changeset
170 {
kono
parents:
diff changeset
171 /* When we get to the end of a dimension, reset it and increment
kono
parents:
diff changeset
172 the next dimension. */
kono
parents:
diff changeset
173 count[n] = 0;
kono
parents:
diff changeset
174 /* We could precalculate these products, but this is a less
kono
parents:
diff changeset
175 frequently used path so probably not worth it. */
kono
parents:
diff changeset
176 base -= sstride[n] * extent[n];
kono
parents:
diff changeset
177 dest -= dstride[n] * extent[n];
kono
parents:
diff changeset
178 n++;
kono
parents:
diff changeset
179 if (n >= rank)
kono
parents:
diff changeset
180 {
kono
parents:
diff changeset
181 /* Break out of the loop. */
kono
parents:
diff changeset
182 continue_loop = 0;
kono
parents:
diff changeset
183 break;
kono
parents:
diff changeset
184 }
kono
parents:
diff changeset
185 else
kono
parents:
diff changeset
186 {
kono
parents:
diff changeset
187 count[n]++;
kono
parents:
diff changeset
188 base += sstride[n];
kono
parents:
diff changeset
189 dest += dstride[n];
kono
parents:
diff changeset
190 }
kono
parents:
diff changeset
191 }
kono
parents:
diff changeset
192 }
kono
parents:
diff changeset
193 }
kono
parents:
diff changeset
194
kono
parents:
diff changeset
195
kono
parents:
diff changeset
196 extern void msum_i2 (gfc_array_i2 * const restrict,
kono
parents:
diff changeset
197 gfc_array_i2 * const restrict, const index_type * const restrict,
kono
parents:
diff changeset
198 gfc_array_l1 * const restrict);
kono
parents:
diff changeset
199 export_proto(msum_i2);
kono
parents:
diff changeset
200
kono
parents:
diff changeset
201 void
kono
parents:
diff changeset
202 msum_i2 (gfc_array_i2 * const restrict retarray,
kono
parents:
diff changeset
203 gfc_array_i2 * const restrict array,
kono
parents:
diff changeset
204 const index_type * const restrict pdim,
kono
parents:
diff changeset
205 gfc_array_l1 * const restrict mask)
kono
parents:
diff changeset
206 {
kono
parents:
diff changeset
207 index_type count[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
208 index_type extent[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
209 index_type sstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
210 index_type dstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
211 index_type mstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
212 GFC_INTEGER_2 * restrict dest;
kono
parents:
diff changeset
213 const GFC_INTEGER_2 * restrict base;
kono
parents:
diff changeset
214 const GFC_LOGICAL_1 * restrict mbase;
kono
parents:
diff changeset
215 index_type rank;
kono
parents:
diff changeset
216 index_type dim;
kono
parents:
diff changeset
217 index_type n;
kono
parents:
diff changeset
218 index_type len;
kono
parents:
diff changeset
219 index_type delta;
kono
parents:
diff changeset
220 index_type mdelta;
kono
parents:
diff changeset
221 int mask_kind;
kono
parents:
diff changeset
222
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
223 if (mask == NULL)
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
224 {
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
225 #ifdef HAVE_BACK_ARG
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
226 sum_i2 (retarray, array, pdim, back);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
227 #else
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
228 sum_i2 (retarray, array, pdim);
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
229 #endif
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
230 return;
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
231 }
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
232
111
kono
parents:
diff changeset
233 dim = (*pdim) - 1;
kono
parents:
diff changeset
234 rank = GFC_DESCRIPTOR_RANK (array) - 1;
kono
parents:
diff changeset
235
kono
parents:
diff changeset
236
kono
parents:
diff changeset
237 if (unlikely (dim < 0 || dim > rank))
kono
parents:
diff changeset
238 {
kono
parents:
diff changeset
239 runtime_error ("Dim argument incorrect in SUM intrinsic: "
kono
parents:
diff changeset
240 "is %ld, should be between 1 and %ld",
kono
parents:
diff changeset
241 (long int) dim + 1, (long int) rank + 1);
kono
parents:
diff changeset
242 }
kono
parents:
diff changeset
243
kono
parents:
diff changeset
244 len = GFC_DESCRIPTOR_EXTENT(array,dim);
kono
parents:
diff changeset
245 if (len <= 0)
kono
parents:
diff changeset
246 return;
kono
parents:
diff changeset
247
kono
parents:
diff changeset
248 mbase = mask->base_addr;
kono
parents:
diff changeset
249
kono
parents:
diff changeset
250 mask_kind = GFC_DESCRIPTOR_SIZE (mask);
kono
parents:
diff changeset
251
kono
parents:
diff changeset
252 if (mask_kind == 1 || mask_kind == 2 || mask_kind == 4 || mask_kind == 8
kono
parents:
diff changeset
253 #ifdef HAVE_GFC_LOGICAL_16
kono
parents:
diff changeset
254 || mask_kind == 16
kono
parents:
diff changeset
255 #endif
kono
parents:
diff changeset
256 )
kono
parents:
diff changeset
257 mbase = GFOR_POINTER_TO_L1 (mbase, mask_kind);
kono
parents:
diff changeset
258 else
kono
parents:
diff changeset
259 runtime_error ("Funny sized logical array");
kono
parents:
diff changeset
260
kono
parents:
diff changeset
261 delta = GFC_DESCRIPTOR_STRIDE(array,dim);
kono
parents:
diff changeset
262 mdelta = GFC_DESCRIPTOR_STRIDE_BYTES(mask,dim);
kono
parents:
diff changeset
263
kono
parents:
diff changeset
264 for (n = 0; n < dim; n++)
kono
parents:
diff changeset
265 {
kono
parents:
diff changeset
266 sstride[n] = GFC_DESCRIPTOR_STRIDE(array,n);
kono
parents:
diff changeset
267 mstride[n] = GFC_DESCRIPTOR_STRIDE_BYTES(mask,n);
kono
parents:
diff changeset
268 extent[n] = GFC_DESCRIPTOR_EXTENT(array,n);
kono
parents:
diff changeset
269
kono
parents:
diff changeset
270 if (extent[n] < 0)
kono
parents:
diff changeset
271 extent[n] = 0;
kono
parents:
diff changeset
272
kono
parents:
diff changeset
273 }
kono
parents:
diff changeset
274 for (n = dim; n < rank; n++)
kono
parents:
diff changeset
275 {
kono
parents:
diff changeset
276 sstride[n] = GFC_DESCRIPTOR_STRIDE(array,n + 1);
kono
parents:
diff changeset
277 mstride[n] = GFC_DESCRIPTOR_STRIDE_BYTES(mask, n + 1);
kono
parents:
diff changeset
278 extent[n] = GFC_DESCRIPTOR_EXTENT(array, n + 1);
kono
parents:
diff changeset
279
kono
parents:
diff changeset
280 if (extent[n] < 0)
kono
parents:
diff changeset
281 extent[n] = 0;
kono
parents:
diff changeset
282 }
kono
parents:
diff changeset
283
kono
parents:
diff changeset
284 if (retarray->base_addr == NULL)
kono
parents:
diff changeset
285 {
kono
parents:
diff changeset
286 size_t alloc_size, str;
kono
parents:
diff changeset
287
kono
parents:
diff changeset
288 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
289 {
kono
parents:
diff changeset
290 if (n == 0)
kono
parents:
diff changeset
291 str = 1;
kono
parents:
diff changeset
292 else
kono
parents:
diff changeset
293 str= GFC_DESCRIPTOR_STRIDE(retarray,n-1) * extent[n-1];
kono
parents:
diff changeset
294
kono
parents:
diff changeset
295 GFC_DIMENSION_SET(retarray->dim[n], 0, extent[n] - 1, str);
kono
parents:
diff changeset
296
kono
parents:
diff changeset
297 }
kono
parents:
diff changeset
298
kono
parents:
diff changeset
299 alloc_size = GFC_DESCRIPTOR_STRIDE(retarray,rank-1) * extent[rank-1];
kono
parents:
diff changeset
300
kono
parents:
diff changeset
301 retarray->offset = 0;
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
302 retarray->dtype.rank = rank;
111
kono
parents:
diff changeset
303
kono
parents:
diff changeset
304 if (alloc_size == 0)
kono
parents:
diff changeset
305 {
kono
parents:
diff changeset
306 /* Make sure we have a zero-sized array. */
kono
parents:
diff changeset
307 GFC_DIMENSION_SET(retarray->dim[0], 0, -1, 1);
kono
parents:
diff changeset
308 return;
kono
parents:
diff changeset
309 }
kono
parents:
diff changeset
310 else
kono
parents:
diff changeset
311 retarray->base_addr = xmallocarray (alloc_size, sizeof (GFC_INTEGER_2));
kono
parents:
diff changeset
312
kono
parents:
diff changeset
313 }
kono
parents:
diff changeset
314 else
kono
parents:
diff changeset
315 {
kono
parents:
diff changeset
316 if (rank != GFC_DESCRIPTOR_RANK (retarray))
kono
parents:
diff changeset
317 runtime_error ("rank of return array incorrect in SUM intrinsic");
kono
parents:
diff changeset
318
kono
parents:
diff changeset
319 if (unlikely (compile_options.bounds_check))
kono
parents:
diff changeset
320 {
kono
parents:
diff changeset
321 bounds_ifunction_return ((array_t *) retarray, extent,
kono
parents:
diff changeset
322 "return value", "SUM");
kono
parents:
diff changeset
323 bounds_equal_extents ((array_t *) mask, (array_t *) array,
kono
parents:
diff changeset
324 "MASK argument", "SUM");
kono
parents:
diff changeset
325 }
kono
parents:
diff changeset
326 }
kono
parents:
diff changeset
327
kono
parents:
diff changeset
328 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
329 {
kono
parents:
diff changeset
330 count[n] = 0;
kono
parents:
diff changeset
331 dstride[n] = GFC_DESCRIPTOR_STRIDE(retarray,n);
kono
parents:
diff changeset
332 if (extent[n] <= 0)
kono
parents:
diff changeset
333 return;
kono
parents:
diff changeset
334 }
kono
parents:
diff changeset
335
kono
parents:
diff changeset
336 dest = retarray->base_addr;
kono
parents:
diff changeset
337 base = array->base_addr;
kono
parents:
diff changeset
338
kono
parents:
diff changeset
339 while (base)
kono
parents:
diff changeset
340 {
kono
parents:
diff changeset
341 const GFC_INTEGER_2 * restrict src;
kono
parents:
diff changeset
342 const GFC_LOGICAL_1 * restrict msrc;
kono
parents:
diff changeset
343 GFC_INTEGER_2 result;
kono
parents:
diff changeset
344 src = base;
kono
parents:
diff changeset
345 msrc = mbase;
kono
parents:
diff changeset
346 {
kono
parents:
diff changeset
347
kono
parents:
diff changeset
348 result = 0;
kono
parents:
diff changeset
349 for (n = 0; n < len; n++, src += delta, msrc += mdelta)
kono
parents:
diff changeset
350 {
kono
parents:
diff changeset
351
kono
parents:
diff changeset
352 if (*msrc)
kono
parents:
diff changeset
353 result += *src;
kono
parents:
diff changeset
354 }
kono
parents:
diff changeset
355 *dest = result;
kono
parents:
diff changeset
356 }
kono
parents:
diff changeset
357 /* Advance to the next element. */
kono
parents:
diff changeset
358 count[0]++;
kono
parents:
diff changeset
359 base += sstride[0];
kono
parents:
diff changeset
360 mbase += mstride[0];
kono
parents:
diff changeset
361 dest += dstride[0];
kono
parents:
diff changeset
362 n = 0;
kono
parents:
diff changeset
363 while (count[n] == extent[n])
kono
parents:
diff changeset
364 {
kono
parents:
diff changeset
365 /* When we get to the end of a dimension, reset it and increment
kono
parents:
diff changeset
366 the next dimension. */
kono
parents:
diff changeset
367 count[n] = 0;
kono
parents:
diff changeset
368 /* We could precalculate these products, but this is a less
kono
parents:
diff changeset
369 frequently used path so probably not worth it. */
kono
parents:
diff changeset
370 base -= sstride[n] * extent[n];
kono
parents:
diff changeset
371 mbase -= mstride[n] * extent[n];
kono
parents:
diff changeset
372 dest -= dstride[n] * extent[n];
kono
parents:
diff changeset
373 n++;
kono
parents:
diff changeset
374 if (n >= rank)
kono
parents:
diff changeset
375 {
kono
parents:
diff changeset
376 /* Break out of the loop. */
kono
parents:
diff changeset
377 base = NULL;
kono
parents:
diff changeset
378 break;
kono
parents:
diff changeset
379 }
kono
parents:
diff changeset
380 else
kono
parents:
diff changeset
381 {
kono
parents:
diff changeset
382 count[n]++;
kono
parents:
diff changeset
383 base += sstride[n];
kono
parents:
diff changeset
384 mbase += mstride[n];
kono
parents:
diff changeset
385 dest += dstride[n];
kono
parents:
diff changeset
386 }
kono
parents:
diff changeset
387 }
kono
parents:
diff changeset
388 }
kono
parents:
diff changeset
389 }
kono
parents:
diff changeset
390
kono
parents:
diff changeset
391
kono
parents:
diff changeset
392 extern void ssum_i2 (gfc_array_i2 * const restrict,
kono
parents:
diff changeset
393 gfc_array_i2 * const restrict, const index_type * const restrict,
kono
parents:
diff changeset
394 GFC_LOGICAL_4 *);
kono
parents:
diff changeset
395 export_proto(ssum_i2);
kono
parents:
diff changeset
396
kono
parents:
diff changeset
397 void
kono
parents:
diff changeset
398 ssum_i2 (gfc_array_i2 * const restrict retarray,
kono
parents:
diff changeset
399 gfc_array_i2 * const restrict array,
kono
parents:
diff changeset
400 const index_type * const restrict pdim,
kono
parents:
diff changeset
401 GFC_LOGICAL_4 * mask)
kono
parents:
diff changeset
402 {
kono
parents:
diff changeset
403 index_type count[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
404 index_type extent[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
405 index_type dstride[GFC_MAX_DIMENSIONS];
kono
parents:
diff changeset
406 GFC_INTEGER_2 * restrict dest;
kono
parents:
diff changeset
407 index_type rank;
kono
parents:
diff changeset
408 index_type n;
kono
parents:
diff changeset
409 index_type dim;
kono
parents:
diff changeset
410
kono
parents:
diff changeset
411
145
1830386684a0 gcc-9.2.0
anatofuz
parents: 131
diff changeset
412 if (mask == NULL || *mask)
111
kono
parents:
diff changeset
413 {
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
414 #ifdef HAVE_BACK_ARG
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
415 sum_i2 (retarray, array, pdim, back);
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
416 #else
111
kono
parents:
diff changeset
417 sum_i2 (retarray, array, pdim);
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
418 #endif
111
kono
parents:
diff changeset
419 return;
kono
parents:
diff changeset
420 }
kono
parents:
diff changeset
421 /* Make dim zero based to avoid confusion. */
kono
parents:
diff changeset
422 dim = (*pdim) - 1;
kono
parents:
diff changeset
423 rank = GFC_DESCRIPTOR_RANK (array) - 1;
kono
parents:
diff changeset
424
kono
parents:
diff changeset
425 if (unlikely (dim < 0 || dim > rank))
kono
parents:
diff changeset
426 {
kono
parents:
diff changeset
427 runtime_error ("Dim argument incorrect in SUM intrinsic: "
kono
parents:
diff changeset
428 "is %ld, should be between 1 and %ld",
kono
parents:
diff changeset
429 (long int) dim + 1, (long int) rank + 1);
kono
parents:
diff changeset
430 }
kono
parents:
diff changeset
431
kono
parents:
diff changeset
432 for (n = 0; n < dim; n++)
kono
parents:
diff changeset
433 {
kono
parents:
diff changeset
434 extent[n] = GFC_DESCRIPTOR_EXTENT(array,n);
kono
parents:
diff changeset
435
kono
parents:
diff changeset
436 if (extent[n] <= 0)
kono
parents:
diff changeset
437 extent[n] = 0;
kono
parents:
diff changeset
438 }
kono
parents:
diff changeset
439
kono
parents:
diff changeset
440 for (n = dim; n < rank; n++)
kono
parents:
diff changeset
441 {
kono
parents:
diff changeset
442 extent[n] =
kono
parents:
diff changeset
443 GFC_DESCRIPTOR_EXTENT(array,n + 1);
kono
parents:
diff changeset
444
kono
parents:
diff changeset
445 if (extent[n] <= 0)
kono
parents:
diff changeset
446 extent[n] = 0;
kono
parents:
diff changeset
447 }
kono
parents:
diff changeset
448
kono
parents:
diff changeset
449 if (retarray->base_addr == NULL)
kono
parents:
diff changeset
450 {
kono
parents:
diff changeset
451 size_t alloc_size, str;
kono
parents:
diff changeset
452
kono
parents:
diff changeset
453 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
454 {
kono
parents:
diff changeset
455 if (n == 0)
kono
parents:
diff changeset
456 str = 1;
kono
parents:
diff changeset
457 else
kono
parents:
diff changeset
458 str = GFC_DESCRIPTOR_STRIDE(retarray,n-1) * extent[n-1];
kono
parents:
diff changeset
459
kono
parents:
diff changeset
460 GFC_DIMENSION_SET(retarray->dim[n], 0, extent[n] - 1, str);
kono
parents:
diff changeset
461
kono
parents:
diff changeset
462 }
kono
parents:
diff changeset
463
kono
parents:
diff changeset
464 retarray->offset = 0;
131
84e7813d76e9 gcc-8.2
mir3636
parents: 111
diff changeset
465 retarray->dtype.rank = rank;
111
kono
parents:
diff changeset
466
kono
parents:
diff changeset
467 alloc_size = GFC_DESCRIPTOR_STRIDE(retarray,rank-1) * extent[rank-1];
kono
parents:
diff changeset
468
kono
parents:
diff changeset
469 if (alloc_size == 0)
kono
parents:
diff changeset
470 {
kono
parents:
diff changeset
471 /* Make sure we have a zero-sized array. */
kono
parents:
diff changeset
472 GFC_DIMENSION_SET(retarray->dim[0], 0, -1, 1);
kono
parents:
diff changeset
473 return;
kono
parents:
diff changeset
474 }
kono
parents:
diff changeset
475 else
kono
parents:
diff changeset
476 retarray->base_addr = xmallocarray (alloc_size, sizeof (GFC_INTEGER_2));
kono
parents:
diff changeset
477 }
kono
parents:
diff changeset
478 else
kono
parents:
diff changeset
479 {
kono
parents:
diff changeset
480 if (rank != GFC_DESCRIPTOR_RANK (retarray))
kono
parents:
diff changeset
481 runtime_error ("rank of return array incorrect in"
kono
parents:
diff changeset
482 " SUM intrinsic: is %ld, should be %ld",
kono
parents:
diff changeset
483 (long int) (GFC_DESCRIPTOR_RANK (retarray)),
kono
parents:
diff changeset
484 (long int) rank);
kono
parents:
diff changeset
485
kono
parents:
diff changeset
486 if (unlikely (compile_options.bounds_check))
kono
parents:
diff changeset
487 {
kono
parents:
diff changeset
488 for (n=0; n < rank; n++)
kono
parents:
diff changeset
489 {
kono
parents:
diff changeset
490 index_type ret_extent;
kono
parents:
diff changeset
491
kono
parents:
diff changeset
492 ret_extent = GFC_DESCRIPTOR_EXTENT(retarray,n);
kono
parents:
diff changeset
493 if (extent[n] != ret_extent)
kono
parents:
diff changeset
494 runtime_error ("Incorrect extent in return value of"
kono
parents:
diff changeset
495 " SUM intrinsic in dimension %ld:"
kono
parents:
diff changeset
496 " is %ld, should be %ld", (long int) n + 1,
kono
parents:
diff changeset
497 (long int) ret_extent, (long int) extent[n]);
kono
parents:
diff changeset
498 }
kono
parents:
diff changeset
499 }
kono
parents:
diff changeset
500 }
kono
parents:
diff changeset
501
kono
parents:
diff changeset
502 for (n = 0; n < rank; n++)
kono
parents:
diff changeset
503 {
kono
parents:
diff changeset
504 count[n] = 0;
kono
parents:
diff changeset
505 dstride[n] = GFC_DESCRIPTOR_STRIDE(retarray,n);
kono
parents:
diff changeset
506 }
kono
parents:
diff changeset
507
kono
parents:
diff changeset
508 dest = retarray->base_addr;
kono
parents:
diff changeset
509
kono
parents:
diff changeset
510 while(1)
kono
parents:
diff changeset
511 {
kono
parents:
diff changeset
512 *dest = 0;
kono
parents:
diff changeset
513 count[0]++;
kono
parents:
diff changeset
514 dest += dstride[0];
kono
parents:
diff changeset
515 n = 0;
kono
parents:
diff changeset
516 while (count[n] == extent[n])
kono
parents:
diff changeset
517 {
kono
parents:
diff changeset
518 /* When we get to the end of a dimension, reset it and increment
kono
parents:
diff changeset
519 the next dimension. */
kono
parents:
diff changeset
520 count[n] = 0;
kono
parents:
diff changeset
521 /* We could precalculate these products, but this is a less
kono
parents:
diff changeset
522 frequently used path so probably not worth it. */
kono
parents:
diff changeset
523 dest -= dstride[n] * extent[n];
kono
parents:
diff changeset
524 n++;
kono
parents:
diff changeset
525 if (n >= rank)
kono
parents:
diff changeset
526 return;
kono
parents:
diff changeset
527 else
kono
parents:
diff changeset
528 {
kono
parents:
diff changeset
529 count[n]++;
kono
parents:
diff changeset
530 dest += dstride[n];
kono
parents:
diff changeset
531 }
kono
parents:
diff changeset
532 }
kono
parents:
diff changeset
533 }
kono
parents:
diff changeset
534 }
kono
parents:
diff changeset
535
kono
parents:
diff changeset
536 #endif