mlemanczyk/even-perfect-numbers-scanner
1
0 Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Rented pools made thread static

Browse Source
This commit is contained in:
mlemanczyk committed 2025-10-11 14:14:28 +02:00
1 parent 1cbf49c8b5
commit 12140bca0a
1 file changed
+17 -6
Download Patch File Download Diff File Expand all files Collapse all files
PerfectNumbers.Core/PrimeOrderCalculator.cs
+17 -6
View File
@@ -233,14 +233,24 @@ internal static partial class PrimeOrderCalculator
return order;
}
[ThreadStatic]
private static ArrayPool<FactorEntry>? _factorPool;
private static ArrayPool<FactorEntry> FactorPool => _factorPool ??= ArrayPool<FactorEntry>.Create();
[ThreadStatic]
private static ArrayPool<ulong>? _ulongPool;
private static ArrayPool<ulong> ULongPool => _ulongPool ??= ArrayPool<ulong>.Create();
private static ulong ExponentLoweringCpu(ulong order, ulong prime, PartialFactorResult factors, in MontgomeryDivisorData divisorData)
{
ArrayPool<FactorEntry> factorPool = FactorPool;
FactorEntry[]? tempArray = null;
try
{
ReadOnlySpan<FactorEntry> factorSpan = factors.Factors;
int length = factors.Count;
tempArray = ArrayPool<FactorEntry>.Shared.Rent(length + 1);
tempArray = factorPool.Rent(length + 1);
Span<FactorEntry> buffer = tempArray;
factorSpan.CopyTo(buffer);
@@ -282,7 +292,7 @@ internal static partial class PrimeOrderCalculator
{
if (tempArray is not null)
{
ArrayPool<FactorEntry>.Shared.Return(tempArray, clearArray: false);
factorPool.Return(tempArray, clearArray: false);
}
}
}
@@ -431,6 +441,7 @@ internal static partial class PrimeOrderCalculator
const int MaxGpuBatchSize = 1024;
const int StackGpuBatchSize = 1024;
Span<ulong> stackGpuRemainders = stackalloc ulong[StackGpuBatchSize];
ArrayPool<ulong> uLongPool = ULongPool;
while (index < candidateCount && powUsed < powBudget)
{
int remaining = candidateCount - index;
@@ -462,7 +473,7 @@ internal static partial class PrimeOrderCalculator
}
else
{
gpuPool = ArrayPool<ulong>.Shared.Rent(batchSize);
gpuPool = uLongPool.Rent(batchSize);
Span<ulong> pooledRemainders = gpuPool.AsSpan(0, batchSize);
status = PrimeOrderGpuHeuristics.TryPow2ModBatch(batch, prime, pooledRemainders, divisorData);
if (status == GpuPow2ModStatus.Success)
@@ -472,7 +483,7 @@ internal static partial class PrimeOrderCalculator
}
else
{
ArrayPool<ulong>.Shared.Return(gpuPool, clearArray: false);
uLongPool.Return(gpuPool, clearArray: false);
gpuPool = null;
}
}
@@ -510,7 +521,7 @@ internal static partial class PrimeOrderCalculator
if (gpuPool is not null)
{
ArrayPool<ulong>.Shared.Return(gpuPool, clearArray: false);
uLongPool.Return(gpuPool, clearArray: false);
}
result = candidate;
@@ -519,7 +530,7 @@ internal static partial class PrimeOrderCalculator
if (gpuPool is not null)
{
ArrayPool<ulong>.Shared.Return(gpuPool, clearArray: false);
uLongPool.Return(gpuPool, clearArray: false);
}
index += batchSize;