@@ -83,12 +83,10 @@ contains
8383 select case(dim)
8484 #:for fi in range(1, rank+1)
8585 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))
8788 do j${"_" * fj}$ = 1, size(x, ${fj}$)
8889 #:endfor
89- #:for fj in range(fi+1, rank+1)
90- do j${"_" * fj}$ = 1, size(x, ${fj}$)
91- #:endfor
9290 x_tmp(:) = x${select_subvector('j', rank, fi)}$
9391 call sort(x_tmp)
9492
@@ -177,12 +175,10 @@ contains
177175 select case(dim)
178176 #:for fi in range(1, rank+1)
179177 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))
181180 do j${"_" * fj}$ = 1, size(x, ${fj}$)
182181 #:endfor
183- #:for fj in range(fi+1, rank+1)
184- do j${"_" * fj}$ = 1, size(x, ${fj}$)
185- #:endfor
186182 x_tmp = pack(x${select_subvector('j', rank, fi)}$, &
187183 mask${select_subvector('j', rank, fi)}$)
188184 call sort(x_tmp)
0 commit comments