From d4c23d94f8060e3018bc727220538559f6861088 Mon Sep 17 00:00:00 2001
From: Nicolau Leal Werneck <nwerneck@gmail.com>
Date: Fri, 14 Oct 2022 08:24:34 +0200
Subject: [PATCH] removing extra outer loop on second stage

---
 src/SortingAlgorithms.jl | 36 +++++++++++++++---------------------
 1 file changed, 15 insertions(+), 21 deletions(-)

diff --git a/src/SortingAlgorithms.jl b/src/SortingAlgorithms.jl
index a84c118..64ed633 100644
--- a/src/SortingAlgorithms.jl
+++ b/src/SortingAlgorithms.jl
@@ -655,33 +655,27 @@ end
 
 function bitonicfirststage!(v, ::Val{Gap}, o::Ordering) where Gap
     N = length(v)
-    @inbounds begin
-        gap = 1 << Gap
-        g2 = gap >> 1
-        for i in 0:gap:N-1
-            firstj = max(0, i + gap - N)
-            for j in firstj:(g2-1)
-                ia = i + j
-                ib = i + gap - j - 1
-                @inbounds comparator!(v, ia + 1, ib + 1, o)
-            end
+    gap = 1 << Gap
+    halfgap = 1 << (Gap - 1)
+    for i in 0:gap:N-1
+        firstj = max(0, i + gap - N)
+        for j in firstj:(halfgap-1)
+            ia = i + j
+            ib = i + gap - j - 1
+            @inbounds comparator!(v, ia + 1, ib + 1, o)
         end
     end
 end
 
 function bitonicsecondstage!(v, ::Val{Gap}, o::Ordering) where Gap
     N = length(v)
-    @inbounds begin
-        for n in Gap:-1:1
-            gap = 1 << n
-            for i in 0:gap:N-1
-                lastj = min(N - 1, N - (i + gap >> 1 + 1))
-                for j in 0:lastj
-                    ia = i + j
-                    ib = i + j + gap >> 1
-                    @inbounds comparator!(v, ia + 1, ib + 1, o)
-                end
-            end
+    gap = 1 << Gap
+    for i in 0:gap:N-1
+        lastj = min(N - 1, N - (i + gap >> 1 + 1))
+        for j in 0:lastj
+            ia = i + j
+            ib = i + j + gap >> 1
+            @inbounds comparator!(v, ia + 1, ib + 1, o)
         end
     end
 end