Reduce GC in CPU odd power ladder

This commit is contained in:
Marek Lemańczyk committed 2025-10-11 22:45:46 +02:00
1 parent b33158390f
commit 2da549fe8c
1 file changed
+27 -14
+27 -14
View File
@@ -346,7 +346,26 @@ public static class ULongExtensions
ulong result = divisor.MontgomeryOne;
ulong nPrime = divisor.NPrime;
ulong[] oddPowers = InitializeMontgomeryOddPowersCpu(divisor, modulus, nPrime, oddPowerCount);
Span<ulong> oddPowerStorage;
ulong[]? pooledOddPowers = null;
if (oddPowerCount <= PerfectNumberConstants.MaxOddPowersCount)
{
oddPowerStorage = stackalloc ulong[PerfectNumberConstants.MaxOddPowersCount];
}
else if (oddPowerCount < PerfectNumberConstants.PooledArrayThreshold)
{
oddPowerStorage = new ulong[oddPowerCount];
}
else
{
pooledOddPowers = ArrayPool<ulong>.Shared.Rent(oddPowerCount);
oddPowerStorage = pooledOddPowers;
}
Span<ulong> oddPowers = oddPowerStorage[..oddPowerCount];
InitializeMontgomeryOddPowersCpu(divisor, modulus, nPrime, oddPowers);
int index = bitLength - 1;
while (index >= 0)
@@ -384,9 +403,9 @@ public static class ULongExtensions
index = windowStart - 1;
}
if (oddPowerCount >= PerfectNumberConstants.PooledArrayThreshold)
if (pooledOddPowers is not null)
{
ArrayPool<ulong>.Shared.Return(oddPowers);
ArrayPool<ulong>.Shared.Return(pooledOddPowers);
}
if (keepMontgomery)
@@ -533,29 +552,25 @@ public static class ULongExtensions
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]
private static ulong[] InitializeMontgomeryOddPowersCpu(in MontgomeryDivisorData divisor, ulong modulus, ulong nPrime, int oddPowerCount)
private static void InitializeMontgomeryOddPowersCpu(in MontgomeryDivisorData divisor, ulong modulus, ulong nPrime, Span<ulong> destination)
{
ulong[] oddPowers = oddPowerCount < PerfectNumberConstants.PooledArrayThreshold
? new ulong[oddPowerCount]
: ArrayPool<ulong>.Shared.Rent(oddPowerCount);
int oddPowerCount = destination.Length;
if (oddPowerCount == 0)
{
return oddPowers;
return;
}
Span<ulong> destination = oddPowers.AsSpan(0, oddPowerCount);
bool computedOnGpu = oddPowerCount <= PerfectNumberConstants.MaxOddPowersCount
&& MontgomeryOddPowerGpu.TryCompute(divisor, oddPowerCount, destination);
if (computedOnGpu)
{
return oddPowers;
return;
}
destination[0] = divisor.MontgomeryTwo;
if (oddPowerCount == 1)
{
return oddPowers;
return;
}
ulong square = divisor.MontgomeryTwoSquared;
@@ -565,8 +580,6 @@ public static class ULongExtensions
current = current.MontgomeryMultiply(square, modulus, nPrime);
destination[i] = current;
}
return oddPowers;
}
[MethodImpl(MethodImplOptions.AggressiveInlining)]