@@ -83,12 +83,10 @@ contains
83
83
select case(dim)
84
84
#:for fi in range(1, rank+1)
85
85
case(${fi}$)
86
- #:for fj in range(1, fi)
86
+ ! Loop over every dimension of the array except "dim"
87
+ #:for fj in list(range(1, fi)) + list(range(fi+1, rank+1))
87
88
do j${"_" * fj}$ = 1, size(x, ${fj}$)
88
89
#:endfor
89
- #:for fj in range(fi+1, rank+1)
90
- do j${"_" * fj}$ = 1, size(x, ${fj}$)
91
- #:endfor
92
90
x_tmp(:) = x${select_subvector('j', rank, fi)}$
93
91
call sort(x_tmp)
94
92
@@ -177,12 +175,10 @@ contains
177
175
select case(dim)
178
176
#:for fi in range(1, rank+1)
179
177
case(${fi}$)
180
- #:for fj in range(1, fi)
178
+ ! Loop over every dimension of the array except "dim"
179
+ #:for fj in list(range(1, fi)) + list(range(fi+1, rank+1))
181
180
do j${"_" * fj}$ = 1, size(x, ${fj}$)
182
181
#:endfor
183
- #:for fj in range(fi+1, rank+1)
184
- do j${"_" * fj}$ = 1, size(x, ${fj}$)
185
- #:endfor
186
182
x_tmp = pack(x${select_subvector('j', rank, fi)}$, &
187
183
mask${select_subvector('j', rank, fi)}$)
188
184
call sort(x_tmp)
0 commit comments