59 lines
2.9 KiB
C#
59 lines
2.9 KiB
C#
using System.Runtime.CompilerServices;
|
|
using ILGPU;
|
|
using ILGPU.Runtime;
|
|
|
|
namespace PerfectNumbers.Core.Gpu.Accelerators;
|
|
|
|
internal sealed class LastDigitGpuTables
|
|
{
|
|
private static readonly Accelerator[] _accelerators = AcceleratorPool.Shared.Accelerators;
|
|
private static readonly LastDigitGpuTables[] _sharedTables = new LastDigitGpuTables[PerfectNumberConstants.RollingAccelerators];
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
internal static LastDigitGpuTables EnsureStaticTables(int acceleratorIndex) => _sharedTables[acceleratorIndex];
|
|
|
|
[MethodImpl(MethodImplOptions.AggressiveInlining)]
|
|
internal static void WarmUp(int acceleratorIndex, AcceleratorStream stream) => _sharedTables[acceleratorIndex] = new LastDigitGpuTables(_accelerators[acceleratorIndex], stream);
|
|
|
|
internal LastDigitGpuTables(Accelerator accelerator, AcceleratorStream stream)
|
|
{
|
|
var primesLastOne = DivisorGenerator.SmallPrimesLastOne;
|
|
var primesLastSeven = DivisorGenerator.SmallPrimesLastSeven;
|
|
var primesLastThree = DivisorGenerator.SmallPrimesLastThree;
|
|
var primesLastNine = DivisorGenerator.SmallPrimesLastNine;
|
|
var primesPow2LastOne = DivisorGenerator.SmallPrimesPow2LastOne;
|
|
var primesPow2LastSeven = DivisorGenerator.SmallPrimesPow2LastSeven;
|
|
var primesPow2LastThree = DivisorGenerator.SmallPrimesPow2LastThree;
|
|
var primesPow2LastNine = DivisorGenerator.SmallPrimesPow2LastNine;
|
|
DevicePrimesLastOne = accelerator.Allocate1D(stream, primesLastOne);
|
|
DevicePrimesLastSeven = accelerator.Allocate1D(stream, primesLastSeven);
|
|
DevicePrimesLastThree = accelerator.Allocate1D(stream, primesLastThree);
|
|
DevicePrimesLastNine = accelerator.Allocate1D(stream, primesLastNine);
|
|
DevicePrimesPow2LastOne = accelerator.Allocate1D(stream, primesPow2LastOne);
|
|
DevicePrimesPow2LastSeven = accelerator.Allocate1D(stream, primesPow2LastSeven);
|
|
DevicePrimesPow2LastThree = accelerator.Allocate1D(stream, primesPow2LastThree);
|
|
DevicePrimesPow2LastNine = accelerator.Allocate1D(stream, primesPow2LastNine);
|
|
}
|
|
|
|
internal readonly MemoryBuffer1D<uint, Stride1D.Dense> DevicePrimesLastOne;
|
|
internal readonly MemoryBuffer1D<uint, Stride1D.Dense> DevicePrimesLastSeven;
|
|
internal readonly MemoryBuffer1D<uint, Stride1D.Dense> DevicePrimesLastThree;
|
|
internal readonly MemoryBuffer1D<uint, Stride1D.Dense> DevicePrimesLastNine;
|
|
internal readonly MemoryBuffer1D<ulong, Stride1D.Dense> DevicePrimesPow2LastOne;
|
|
internal readonly MemoryBuffer1D<ulong, Stride1D.Dense> DevicePrimesPow2LastSeven;
|
|
internal readonly MemoryBuffer1D<ulong, Stride1D.Dense> DevicePrimesPow2LastThree;
|
|
internal readonly MemoryBuffer1D<ulong, Stride1D.Dense> DevicePrimesPow2LastNine;
|
|
|
|
internal void Dispose()
|
|
{
|
|
DevicePrimesLastOne.Dispose();
|
|
DevicePrimesLastSeven.Dispose();
|
|
DevicePrimesLastThree.Dispose();
|
|
DevicePrimesLastNine.Dispose();
|
|
DevicePrimesPow2LastOne.Dispose();
|
|
DevicePrimesPow2LastSeven.Dispose();
|
|
DevicePrimesPow2LastThree.Dispose();
|
|
DevicePrimesPow2LastNine.Dispose();
|
|
}
|
|
}
|