Skip to content

Commit 6644140

Browse files
committed
add comment + combined fypp loops
1 parent 4f43ce5 commit 6644140

File tree

1 file changed

+4
-8
lines changed

1 file changed

+4
-8
lines changed

src/stdlib_stats_median.fypp

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)