From 3f587a362943a59455c6b177829118fa87e6b7b0 Mon Sep 17 00:00:00 2001
From: Mounir IDRASSI <mounir.idrassi@idrix.fr>
Date: Wed, 1 Jul 2020 17:11:07 +0200
Subject: Windows: Don't use API for Processor Groups support if there is only
 1 CPU group in the system. This can fix slowness issue observed on some PCs
 with AMD CPUs.

---
 src/Driver/Ntdriver.c | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

(limited to 'src/Driver/Ntdriver.c')

diff --git a/src/Driver/Ntdriver.c b/src/Driver/Ntdriver.c
index 188c6647..957853cd 100644
--- a/src/Driver/Ntdriver.c
+++ b/src/Driver/Ntdriver.c
@@ -4491,7 +4491,7 @@ NTSTATUS TCCompleteDiskIrp (PIRP irp, NTSTATUS status, ULONG_PTR information)
 }
 
 
-size_t GetCpuCount ()
+size_t GetCpuCount (WORD* pGroupCount)
 {
 	size_t cpuCount = 0;
 	if (KeQueryActiveGroupCountPtr && KeQueryActiveProcessorCountExPtr)
@@ -4501,6 +4501,9 @@ size_t GetCpuCount ()
 		{
 			cpuCount += (size_t) KeQueryActiveProcessorCountExPtr (i);
 		}
+
+		if (pGroupCount)
+			*pGroupCount = groupCount;
 	}
 	else
 	{
@@ -4514,6 +4517,9 @@ size_t GetCpuCount ()
 
 			activeCpuMap >>= 1;
 		}
+
+		if (pGroupCount)
+			*pGroupCount = 1;
 	}
 
 	if (cpuCount == 0)
-- 
cgit v1.2.3