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

Restored the Dictionary parameter instead of arrays

Browse Source
This commit is contained in:
mlemanczyk committed 2025-10-21 04:08:48 +02:00
1 parent 8dc7751084
commit ea11db6d0f
1 file changed
+23 -23
Download Patch File Download Diff File Expand all files Collapse all files
PerfectNumbers.Core/PrimeOrderCalculator.Gpu.cs
+23 -23
View File
@@ -13,18 +13,18 @@ internal static partial class PrimeOrderCalculator
private const int GpuSmallPrimeFactorSlots = 64; private const int GpuSmallPrimeFactorSlots = 64;
private static bool TryPopulateSmallPrimeFactorsGpu(ulong value, uint limit, Span<ulong> primeBuffer, Span<int> exponentBuffer, out int factorCount, out ulong remaining) private static bool TryPopulateSmallPrimeFactorsGpu(ulong value, uint limit, Dictionary<ulong, int> counts, out int factorCount, out ulong remaining)
{ {
// primeBufferArray = ThreadStaticPools.UlongPool.Rent(GpuSmallPrimeFactorSlots); var primeBufferArray = ThreadStaticPools.UlongPool.Rent(GpuSmallPrimeFactorSlots);
// exponentBufferArray = ThreadStaticPools.IntPool.Rent(GpuSmallPrimeFactorSlots); var exponentBufferArray = ThreadStaticPools.IntPool.Rent(GpuSmallPrimeFactorSlots);
// Span<ulong> primeBuffer = primeBufferArray.AsSpan(0, GpuSmallPrimeFactorSlots); Span<ulong> primeBuffer = primeBufferArray.AsSpan(0, GpuSmallPrimeFactorSlots);
// Span<int> exponentBuffer = exponentBufferArray.AsSpan(0, GpuSmallPrimeFactorSlots); Span<int> exponentBuffer = exponentBufferArray.AsSpan(0, GpuSmallPrimeFactorSlots);
// primeBuffer.Clear(); // primeBuffer.Clear();
// exponentBuffer.Clear(); // exponentBuffer.Clear();
remaining = value; remaining = value;
// try try
// { {
var lease = GpuKernelPool.GetKernel(useGpuOrder: true); var lease = GpuKernelPool.GetKernel(useGpuOrder: true);
var execution = lease.EnterExecutionScope(); var execution = lease.EnterExecutionScope();
try try
@@ -64,19 +64,19 @@ internal static partial class PrimeOrderCalculator
scratch.RemainingSlot.View.CopyToCPU(ref remaining, 1); scratch.RemainingSlot.View.CopyToCPU(ref remaining, 1);
// for (int i = 0; i < factorCount; i++) for (int i = 0; i < factorCount; i++)
{
ulong primeValue = primeBuffer[i];
int exponent = exponentBuffer[i];
// This will never happen in production code
// if (primeValue == 0UL || exponent == 0)
// { // {
// ulong primeValue = primeBuffer[i]; // continue;
// int exponent = exponentBuffer[i];
// // This will never happen in production code
// // if (primeValue == 0UL || exponent == 0)
// // {
// // continue;
// // }
// counts[primeValue] = exponent;
// } // }
counts.Add(primeValue, exponent);
}
return true; return true;
} }
finally finally
@@ -84,11 +84,11 @@ internal static partial class PrimeOrderCalculator
execution.Dispose(); execution.Dispose();
lease.Dispose(); lease.Dispose();
} }
// } }
// finally finally
// { {
// ThreadStaticPools.UlongPool.Return(primeBufferArray); ThreadStaticPools.UlongPool.Return(primeBufferArray);
// ThreadStaticPools.IntPool.Return(exponentBufferArray); ThreadStaticPools.IntPool.Return(exponentBufferArray);
// } }
} }
} }