Reuse GPU context for nested partial factoring

PerfectNumbers.Core/PrimeOrderCalculator.Gpu.cs

@@ -1,5 +1,4 @@
 using System;
-using System.Runtime.InteropServices;
 using ILGPU;
 using ILGPU.Runtime;
 using ILGPU.Runtime.OpenCL;
@@ -32,52 +31,39 @@ internal static partial class PrimeOrderCalculator
 				Accelerator accelerator = lease.Accelerator;
 				AcceleratorStream stream = lease.Stream;
 
-				SmallPrimeFactorTables tables = GpuKernelPool.EnsureSmallPrimeFactorTables(accelerator);
 				SmallPrimeFactorScratch scratch = GpuKernelPool.EnsureSmallPrimeFactorScratch(accelerator, GpuSmallPrimeFactorSlots);
-				// scratch.Clear();
 
-				var kernel = lease.SmallPrimeFactorKernel;
-				kernel(
-					stream,
-					1,
-					value,
-					limit,
-					tables.PrimesView,
-					tables.SquaresView,
-					tables.Count,
-					scratch.PrimeSlots.View,
-					scratch.ExponentSlots.View,
-					scratch.CountSlot.View,
-					scratch.RemainingSlot.View);
+                                bool success = PrimeOrderGpuHeuristics.TryPartialFactor(
+                                        accelerator,
+                                        stream,
+                                        scratch,
+                                        value,
+                                        limit,
+                                        primeBuffer,
+                                        exponentBuffer,
+                                        out factorCount,
+                                        out remaining,
+                                        out _);
 
-				stream.Synchronize();
+                                if (!success)
+                                {
+                                        return false;
+                                }
 
-				factorCount = 0;
-				scratch.CountSlot.View.CopyToCPU(ref factorCount, 1);
-				factorCount = Math.Min(factorCount, GpuSmallPrimeFactorSlots);
+                                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)
+                                        // {
+                                        //      continue;
+                                        // }
 
-				if (factorCount > 0)
-				{
-					scratch.PrimeSlots.View.CopyToCPU(ref MemoryMarshal.GetReference(primeBuffer), factorCount);
-					scratch.ExponentSlots.View.CopyToCPU(ref MemoryMarshal.GetReference(exponentBuffer), factorCount);
-				}
+                                        counts.Add(primeValue, exponent);
+                                }
 
-				scratch.RemainingSlot.View.CopyToCPU(ref remaining, 1);
-
-				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)
-					// {
-					// 	continue;
-					// }
-
-					counts.Add(primeValue, exponent);
-				}
-
-				return true;
+                                return true;
 			}
 			finally
 			{