Stabilization improvements. Stream will be released when releasing GPU lease to prevent application crashes.

PerfectNumbers.Core/Gpu/AcceleratorPool.cs

@@ -3,21 +3,21 @@ using ILGPU.Runtime;
 
 namespace PerfectNumbers.Core.Gpu
 {
-    public struct AcceleratorPool(int capacity)
+	public struct AcceleratorPool
 	{
-		internal readonly Accelerator[] Accelerators = [.. Enumerable.Range(1, capacity).Select(_ => {
-			return SharedGpuContext.CreateAccelerator();
-		})];
+		internal readonly Context Context;
+		internal readonly Device Device;
+		internal readonly Accelerator[] Accelerators;
 
 		private int _index;
 
 		public Accelerator Rent()
 		{
-			Atomic.CompareExchange(ref _index, capacity, 0);
+			Atomic.CompareExchange(ref _index, Capacity, 0);
 			var index = Atomic.Add(ref _index, 1);
-			if (index >= capacity)
+			if (index >= Capacity)
 			{
-				index -= capacity;
+				index -= Capacity;
 			}
 
 			return Accelerators[index];
@@ -29,5 +29,16 @@ namespace PerfectNumbers.Core.Gpu
 		}
 
 		public static readonly AcceleratorPool Shared = new(PerfectNumberConstants.RollingAccelerators);
+		private readonly int Capacity;
+
+		public AcceleratorPool(int capacity)
+		{
+			Capacity = capacity;
+			var context = Context.CreateDefault();
+			var device = context.GetPreferredDevice(false);
+			Context = context;
+			Device = device;
+			Accelerators = [.. Enumerable.Range(0, capacity).Select(_ => device.CreateAccelerator(context))];
+		}
 	}
 }

PerfectNumbers.Core/Gpu/Accelerators/Pow2MontgomeryAccelerator.cs

@@ -8,7 +8,7 @@ namespace PerfectNumbers.Core.Gpu.Accelerators;
 public sealed class Pow2MontgomeryAccelerator
 {
 	#region Pool
-	
+
 	[ThreadStatic]
 	private static Queue<Pow2MontgomeryAccelerator>? _pool;
 
@@ -21,7 +21,11 @@ public sealed class Pow2MontgomeryAccelerator
 		if (!pool.TryDequeue(out var gpu))
 		{
 			Accelerator accelerator = _accelerators.Rent();
-			gpu = new(accelerator);
+			gpu = new(accelerator, PerfectNumberConstants.DefaultFactorsBuffer);
+		}
+		else
+		{
+			gpu.Stream = gpu.Accelerator.CreateStream();
 		}
 
 		return gpu;
@@ -40,17 +44,33 @@ public sealed class Pow2MontgomeryAccelerator
 	public readonly Accelerator Accelerator;
 	public readonly MemoryBuffer1D<ulong, Stride1D.Dense> Input;
 	public readonly MemoryBuffer1D<ulong, Stride1D.Dense> Output;
+	public MemoryBuffer1D<KeyValuePair<ulong, int>, Stride1D.Dense> ToTestOnDevice;
+	public Dictionary<ulong, int> ToTestOnHost = [];
 	public AcceleratorStream? Stream;
 
 	public readonly Action<AcceleratorStream, Index1D, ArrayView1D<ulong, Stride1D.Dense>, MontgomeryDivisorData, ArrayView1D<ulong, Stride1D.Dense>> ConvertToStandardKernel;
 	public readonly Action<AcceleratorStream, Index1D, ArrayView1D<ulong, Stride1D.Dense>, MontgomeryDivisorData, ArrayView1D<ulong, Stride1D.Dense>> KeepMontgomeryKernel;
 
-	public Pow2MontgomeryAccelerator(Accelerator accelerator)
+	public readonly Kernel CheckFactorsKernel;
+
+	public void EnsureCapacity(int factorsCount)
+	{
+		if (factorsCount > ToTestOnDevice.Length)
+		{
+			Accelerator accelerator = Accelerator;
+			ToTestOnDevice.Dispose();
+			ToTestOnDevice = accelerator.Allocate1D<KeyValuePair<ulong, int>>(factorsCount);
+		}
+	}
+
+	public Pow2MontgomeryAccelerator(Accelerator accelerator, int factorsCount)
 	{
 		Accelerator = accelerator;
+		Stream = accelerator.CreateStream();
+
 		Input = accelerator.Allocate1D<ulong>(1);
 		Output = accelerator.Allocate1D<ulong>(1);
-		Stream = accelerator.CreateStream();
+		ToTestOnDevice = accelerator.Allocate1D<KeyValuePair<ulong, int>>(factorsCount);
 
 		var keepKernel = accelerator.LoadAutoGroupedStreamKernel<Index1D, ArrayView1D<ulong, Stride1D.Dense>, MontgomeryDivisorData, ArrayView1D<ulong, Stride1D.Dense>>(Pow2MontgomeryKernels.Pow2MontgomeryKernelKeepMontgomery);
 		var keepLauncher = KernelUtil.GetKernel(keepKernel);
@@ -61,6 +81,9 @@ public sealed class Pow2MontgomeryAccelerator
 		var convertLauncher = KernelUtil.GetKernel(convertKernel);
 
 		ConvertToStandardKernel = convertLauncher.CreateLauncherDelegate<Action<AcceleratorStream, Index1D, ArrayView1D<ulong, Stride1D.Dense>, MontgomeryDivisorData, ArrayView1D<ulong, Stride1D.Dense>>>();
+
+		var checkFactorsKernel = accelerator.LoadStreamKernel<int, ulong, ArrayView1D<KeyValuePair<ulong, int>, Stride1D.Dense>, MontgomeryDivisorData, ArrayView1D<ulong, Stride1D.Dense>>(PrimeOrderGpuHeuristics.CheckFactorsKernel);
+		CheckFactorsKernel = KernelUtil.GetKernel(checkFactorsKernel);
 	}
 
 	public void Dispose()
@@ -68,7 +91,8 @@ public sealed class Pow2MontgomeryAccelerator
 		Input.Dispose();
 		Output.Dispose();
 		Stream?.Dispose();
+		ToTestOnDevice.Dispose();
 		// Accelerator is shared with other threads
 		// Accelerator.Dispose();
 	}
-}
+}