From 01f0f0236a8524da66b54bf438e16514ed0ffe8e Mon Sep 17 00:00:00 2001
From: Raghuveer Devulapalli <raghuveer.devulapalli@intel.com>
Date: Thu, 12 Oct 2023 14:53:07 -0700
Subject: [PATCH] Use num_unroll = 9 for 16-bit data

---
 src/avx512-16bit-qsort.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/avx512-16bit-qsort.hpp b/src/avx512-16bit-qsort.hpp
index 820a4440..fdfba924 100644
--- a/src/avx512-16bit-qsort.hpp
+++ b/src/avx512-16bit-qsort.hpp
@@ -25,7 +25,7 @@ struct zmm_vector<float16> {
 #else
     static constexpr int network_sort_threshold = 512;
 #endif
-    static constexpr int partition_unroll_factor = 0;
+    static constexpr int partition_unroll_factor = 8;
 
     using swizzle_ops = avx512_16bit_swizzle_ops;
 
@@ -191,7 +191,7 @@ struct zmm_vector<int16_t> {
 #else
     static constexpr int network_sort_threshold = 512;
 #endif
-    static constexpr int partition_unroll_factor = 0;
+    static constexpr int partition_unroll_factor = 8;
 
     using swizzle_ops = avx512_16bit_swizzle_ops;
 
@@ -314,7 +314,7 @@ struct zmm_vector<uint16_t> {
 #else
     static constexpr int network_sort_threshold = 512;
 #endif
-    static constexpr int partition_unroll_factor = 0;
+    static constexpr int partition_unroll_factor = 8;
 
     using swizzle_ops = avx512_16bit_swizzle_ops;