Move by-divisor tests to large-cycle divisors
This commit is contained in:
1 parent
c0b6b29837
commit
594a38b9d7
2 files changed
+120
-166
No files matched your search
@@ -1,8 +1,6 @@
|
|||||||
using System;
|
using System;
|
||||||
using FluentAssertions;
|
using FluentAssertions;
|
||||||
using PerfectNumbers.Core;
|
using PerfectNumbers.Core;
|
||||||
using System.Globalization;
|
|
||||||
using System.Reflection;
|
|
||||||
using PerfectNumbers.Core.Cpu;
|
using PerfectNumbers.Core.Cpu;
|
||||||
using Xunit;
|
using Xunit;
|
||||||
|
|
||||||
@@ -10,20 +8,23 @@ namespace PerfectNumbers.Core.Tests;
|
|||||||
|
|
||||||
public class MersenneNumberDivisorCpuTesterTests
|
public class MersenneNumberDivisorCpuTesterTests
|
||||||
{
|
{
|
||||||
|
private const ulong LargeDivisor1 = 4_000_169UL;
|
||||||
|
private const ulong LargeDivisor2 = 4_000_561UL;
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_tester_tracks_divisors_across_primes()
|
public void ByDivisor_tester_tracks_divisors_across_primes()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
||||||
tester.ConfigureFromMaxPrime(11UL);
|
tester.ConfigureFromMaxPrime(83UL);
|
||||||
|
|
||||||
RunIsPrimeWithSnapshot(tester, 5UL, out bool divisorsExhausted);
|
tester.IsPrime(59UL, out bool divisorsExhausted).Should().BeTrue();
|
||||||
divisorsExhausted.Should().BeTrue();
|
divisorsExhausted.Should().BeTrue();
|
||||||
|
|
||||||
RunIsPrimeWithSnapshot(tester, 7UL, out divisorsExhausted);
|
tester.IsPrime(61UL, out divisorsExhausted).Should().BeTrue();
|
||||||
divisorsExhausted.Should().BeTrue();
|
divisorsExhausted.Should().BeTrue();
|
||||||
|
|
||||||
RunIsPrimeWithSnapshot(tester, 11UL, out divisorsExhausted);
|
tester.IsPrime(67UL, out divisorsExhausted).Should().BeTrue();
|
||||||
divisorsExhausted.Should().BeTrue();
|
divisorsExhausted.Should().BeTrue();
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -32,63 +33,63 @@ public class MersenneNumberDivisorCpuTesterTests
|
|||||||
public void ByDivisor_prime_test_limit_disables_divisor_exhaustion()
|
public void ByDivisor_prime_test_limit_disables_divisor_exhaustion()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
||||||
tester.ConfigureFromMaxPrime(11UL);
|
tester.ConfigureFromMaxPrime(83UL);
|
||||||
|
|
||||||
RunIsPrimeWithSnapshot(tester, 5UL, out bool divisorsExhausted, TimeSpan.Zero);
|
tester.IsPrime(59UL, out bool divisorsExhausted, TimeSpan.Zero).Should().BeTrue();
|
||||||
divisorsExhausted.Should().BeFalse();
|
divisorsExhausted.Should().BeFalse();
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_checks_divisors_across_primes()
|
public void ByDivisor_session_checks_divisors_across_exponents()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
||||||
tester.ConfigureFromMaxPrime(173UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] primes = { 59UL, 61UL, 67UL, 71UL, 73UL };
|
ulong[] exponents = { 59UL, 61UL, 67UL, 71UL, 73UL, 79UL, 117_652UL, 333_380UL };
|
||||||
byte[] hits = new byte[primes.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
MontgomeryDivisorData divisor223 = MontgomeryDivisorData.FromModulus(223UL);
|
MontgomeryDivisorData divisor1 = MontgomeryDivisorData.FromModulus(LargeDivisor1);
|
||||||
ulong cycle223 = MersenneDivisorCycles.CalculateCycleLength(223UL, divisor223);
|
ulong cycle1 = MersenneDivisorCycles.CalculateCycleLength(LargeDivisor1, divisor1);
|
||||||
session.CheckDivisor(223UL, divisor223, cycle223, primes, hits);
|
session.CheckDivisor(LargeDivisor1, divisor1, cycle1, exponents, hits);
|
||||||
|
|
||||||
byte[] expected223 = new byte[primes.Length];
|
byte[] expected1 = new byte[exponents.Length];
|
||||||
for (int i = 0; i < primes.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
{
|
{
|
||||||
expected223[i] = primes[i].Pow2ModWindowedCpu(divisor223.Modulus) == 1UL ? (byte)1 : (byte)0;
|
expected1[i] = exponents[i].Pow2ModWindowedCpu(divisor1.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
}
|
}
|
||||||
|
|
||||||
hits.Should().Equal(expected223);
|
hits.Should().Equal(expected1);
|
||||||
|
|
||||||
Array.Fill(hits, (byte)0);
|
Array.Fill(hits, (byte)0);
|
||||||
MontgomeryDivisorData divisor431 = MontgomeryDivisorData.FromModulus(431UL);
|
MontgomeryDivisorData divisor2 = MontgomeryDivisorData.FromModulus(LargeDivisor2);
|
||||||
ulong cycle431 = MersenneDivisorCycles.CalculateCycleLength(431UL, divisor431);
|
ulong cycle2 = MersenneDivisorCycles.CalculateCycleLength(LargeDivisor2, divisor2);
|
||||||
session.CheckDivisor(431UL, divisor431, cycle431, primes, hits);
|
session.CheckDivisor(LargeDivisor2, divisor2, cycle2, exponents, hits);
|
||||||
|
|
||||||
byte[] expected431 = new byte[primes.Length];
|
byte[] expected2 = new byte[exponents.Length];
|
||||||
for (int i = 0; i < primes.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
{
|
{
|
||||||
expected431[i] = primes[i].Pow2ModWindowedCpu(divisor431.Modulus) == 1UL ? (byte)1 : (byte)0;
|
expected2[i] = exponents[i].Pow2ModWindowedCpu(divisor2.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
}
|
}
|
||||||
|
|
||||||
hits.Should().Equal(expected431);
|
hits.Should().Equal(expected2);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_223_as_composite()
|
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_large_divisor1_as_composite()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
||||||
tester.ConfigureFromMaxPrime(173UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] exponents = { 59UL, 67UL, 118UL, 134UL, 177UL };
|
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(LargeDivisor1);
|
||||||
byte[] hits = new byte[exponents.Length];
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(LargeDivisor1, divisorData);
|
||||||
|
|
||||||
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(223UL);
|
ulong[] exponents = { 59UL, 67UL, cycle, cycle * 2UL, cycle + 1UL };
|
||||||
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(223UL, divisorData);
|
byte[] hits = new byte[exponents.Length];
|
||||||
session.CheckDivisor(223UL, divisorData, cycle, exponents, hits);
|
session.CheckDivisor(LargeDivisor1, divisorData, cycle, exponents, hits);
|
||||||
|
|
||||||
byte[] expected = new byte[exponents.Length];
|
byte[] expected = new byte[exponents.Length];
|
||||||
for (int i = 0; i < exponents.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
@@ -101,18 +102,18 @@ public class MersenneNumberDivisorCpuTesterTests
|
|||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_431_as_composite()
|
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_large_divisor2_as_composite()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
var tester = new MersenneNumberDivisorByDivisorCpuTester();
|
||||||
tester.ConfigureFromMaxPrime(193UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] exponents = { 67UL, 71UL, 86UL, 134UL, 172UL };
|
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(LargeDivisor2);
|
||||||
byte[] hits = new byte[exponents.Length];
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(LargeDivisor2, divisorData);
|
||||||
|
|
||||||
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(431UL);
|
ulong[] exponents = { 61UL, 73UL, cycle, cycle * 2UL, cycle - 1UL };
|
||||||
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(431UL, divisorData);
|
byte[] hits = new byte[exponents.Length];
|
||||||
session.CheckDivisor(431UL, divisorData, cycle, exponents, hits);
|
session.CheckDivisor(LargeDivisor2, divisorData, cycle, exponents, hits);
|
||||||
|
|
||||||
byte[] expected = new byte[exponents.Length];
|
byte[] expected = new byte[exponents.Length];
|
||||||
for (int i = 0; i < exponents.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
@@ -122,50 +123,4 @@ public class MersenneNumberDivisorCpuTesterTests
|
|||||||
|
|
||||||
hits.Should().Equal(expected);
|
hits.Should().Equal(expected);
|
||||||
}
|
}
|
||||||
private static bool RunIsPrimeWithSnapshot(MersenneNumberDivisorByDivisorCpuTester tester, ulong prime, out bool divisorsExhausted, TimeSpan? timeLimit = null)
|
|
||||||
{
|
|
||||||
while (true)
|
|
||||||
{
|
|
||||||
try
|
|
||||||
{
|
|
||||||
return tester.IsPrime(prime, out divisorsExhausted, timeLimit);
|
|
||||||
}
|
}
|
||||||
catch (InvalidDataException ex) when (TryPopulateMissingCycle(ex))
|
|
||||||
{
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
private static bool TryPopulateMissingCycle(InvalidDataException ex)
|
|
||||||
{
|
|
||||||
const string prefix = "Divisor cycle is missing for ";
|
|
||||||
if (!ex.Message.StartsWith(prefix, StringComparison.Ordinal))
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
string suffix = ex.Message[prefix.Length..];
|
|
||||||
if (!ulong.TryParse(suffix, NumberStyles.Integer, CultureInfo.InvariantCulture, out ulong divisor))
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
var cache = DivisorCycleCache.Shared;
|
|
||||||
var snapshotField = typeof(DivisorCycleCache).GetField("_snapshot", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
|
||||||
ulong[] snapshot = (ulong[])snapshotField.GetValue(cache)!;
|
|
||||||
if (divisor >= (ulong)snapshot.Length)
|
|
||||||
{
|
|
||||||
return false;
|
|
||||||
}
|
|
||||||
|
|
||||||
if (snapshot[divisor] != 0UL)
|
|
||||||
{
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
|
|
||||||
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(divisor);
|
|
||||||
snapshot[divisor] = MersenneDivisorCycles.CalculateCycleLength(divisor, divisorData);
|
|
||||||
return true;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
@@ -12,12 +12,12 @@ namespace PerfectNumbers.Core.Tests;
|
|||||||
public class MersenneNumberDivisorGpuTesterTests
|
public class MersenneNumberDivisorGpuTesterTests
|
||||||
{
|
{
|
||||||
[Theory]
|
[Theory]
|
||||||
[InlineData(33UL, 7UL, true)]
|
[InlineData(117_652UL, 4_000_169UL, true)]
|
||||||
[InlineData(35UL, 31UL, true)]
|
[InlineData(333_380UL, 4_000_561UL, true)]
|
||||||
[InlineData(37UL, 223UL, true)]
|
[InlineData(59UL, 4_000_169UL, false)]
|
||||||
[InlineData(33UL, 13UL, false)]
|
[InlineData(61UL, 4_000_561UL, false)]
|
||||||
[InlineData(35UL, 73UL, false)]
|
[InlineData(333_379UL, 4_000_561UL, false)]
|
||||||
[InlineData(37UL, 227UL, false)]
|
[InlineData(117_653UL, 4_000_169UL, false)]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void IsDivisible_returns_expected(ulong exponent, ulong divisor, bool expected)
|
public void IsDivisible_returns_expected(ulong exponent, ulong divisor, bool expected)
|
||||||
{
|
{
|
||||||
@@ -152,23 +152,39 @@ public class MersenneNumberDivisorGpuTesterTests
|
|||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_checks_divisors_across_primes()
|
public void ByDivisor_session_checks_divisors_across_exponents()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorGpuTester();
|
var tester = new MersenneNumberDivisorByDivisorGpuTester();
|
||||||
tester.ConfigureFromMaxPrime(43UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] primes = { 31UL, 37UL, 41UL, 43UL };
|
ulong[] exponents = { 59UL, 61UL, 67UL, 71UL, 73UL, 79UL, 117_652UL, 333_380UL };
|
||||||
byte[] hits = new byte[primes.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
ulong cycle223 = MersenneDivisorCycles.CalculateCycleLength(223UL, MontgomeryDivisorData.FromModulus(223UL));
|
MontgomeryDivisorData divisor1 = MontgomeryDivisorData.FromModulus(4_000_169UL);
|
||||||
session.CheckDivisor(223UL, MontgomeryDivisorData.FromModulus(223UL), cycle223, primes, hits);
|
ulong cycle1 = MersenneDivisorCycles.CalculateCycleLength(4_000_169UL, divisor1);
|
||||||
hits.Should().ContainInOrder(new byte[] { 0, 1, 0, 0 });
|
session.CheckDivisor(4_000_169UL, divisor1, cycle1, exponents, hits);
|
||||||
|
|
||||||
|
byte[] expected1 = new byte[exponents.Length];
|
||||||
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
|
{
|
||||||
|
expected1[i] = exponents[i].Pow2ModWindowedCpu(divisor1.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
|
}
|
||||||
|
|
||||||
|
hits.Should().Equal(expected1);
|
||||||
|
|
||||||
Array.Fill(hits, (byte)0);
|
Array.Fill(hits, (byte)0);
|
||||||
ulong cycle13367 = MersenneDivisorCycles.CalculateCycleLength(13367UL, MontgomeryDivisorData.FromModulus(13367UL));
|
MontgomeryDivisorData divisor2 = MontgomeryDivisorData.FromModulus(4_000_561UL);
|
||||||
session.CheckDivisor(13367UL, MontgomeryDivisorData.FromModulus(13367UL), cycle13367, primes, hits);
|
ulong cycle2 = MersenneDivisorCycles.CalculateCycleLength(4_000_561UL, divisor2);
|
||||||
hits.Should().ContainInOrder(new byte[] { 0, 0, 1, 0 });
|
session.CheckDivisor(4_000_561UL, divisor2, cycle2, exponents, hits);
|
||||||
|
|
||||||
|
byte[] expected2 = new byte[exponents.Length];
|
||||||
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
|
{
|
||||||
|
expected2[i] = exponents[i].Pow2ModWindowedCpu(divisor2.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
|
}
|
||||||
|
|
||||||
|
hits.Should().Equal(expected2);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
@@ -180,43 +196,43 @@ public class MersenneNumberDivisorGpuTesterTests
|
|||||||
GpuBatchSize = 2,
|
GpuBatchSize = 2,
|
||||||
};
|
};
|
||||||
|
|
||||||
tester.ConfigureFromMaxPrime(173UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] primes = { 59UL, 61UL, 67UL, 71UL, 73UL };
|
ulong[] exponents = { 59UL, 61UL, 67UL, 71UL, 73UL, 79UL, 117_652UL };
|
||||||
byte[] hits = new byte[primes.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
MontgomeryDivisorData divisor223 = MontgomeryDivisorData.FromModulus(223UL);
|
MontgomeryDivisorData divisor = MontgomeryDivisorData.FromModulus(4_000_169UL);
|
||||||
ulong cycle223 = MersenneDivisorCycles.CalculateCycleLength(223UL, divisor223);
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(4_000_169UL, divisor);
|
||||||
session.CheckDivisor(223UL, divisor223, cycle223, primes, hits);
|
session.CheckDivisor(4_000_169UL, divisor, cycle, exponents, hits);
|
||||||
|
|
||||||
byte[] expected223 = new byte[primes.Length];
|
byte[] expected = new byte[exponents.Length];
|
||||||
for (int i = 0; i < primes.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
{
|
{
|
||||||
expected223[i] = primes[i].Pow2ModWindowedCpu(divisor223.Modulus) == 1UL ? (byte)1 : (byte)0;
|
expected[i] = exponents[i].Pow2ModWindowedCpu(divisor.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
}
|
}
|
||||||
|
|
||||||
hits.Should().Equal(expected223);
|
hits.Should().Equal(expected);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_223_as_composite()
|
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_large_divisor1_as_composite()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorGpuTester
|
var tester = new MersenneNumberDivisorByDivisorGpuTester
|
||||||
{
|
{
|
||||||
GpuBatchSize = 5,
|
GpuBatchSize = 5,
|
||||||
};
|
};
|
||||||
tester.ConfigureFromMaxPrime(173UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] exponents = { 59UL, 67UL, 118UL, 134UL, 177UL };
|
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(4_000_169UL);
|
||||||
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(4_000_169UL, divisorData);
|
||||||
|
|
||||||
|
ulong[] exponents = { 59UL, 67UL, cycle, cycle * 2UL, cycle + 1UL };
|
||||||
byte[] hits = new byte[exponents.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(223UL);
|
session.CheckDivisor(4_000_169UL, divisorData, cycle, exponents, hits);
|
||||||
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(223UL, divisorData);
|
|
||||||
|
|
||||||
session.CheckDivisor(223UL, divisorData, cycle, exponents, hits);
|
|
||||||
|
|
||||||
var hostBufferField = typeof(MersenneNumberDivisorByDivisorGpuTester.DivisorScanSession)
|
var hostBufferField = typeof(MersenneNumberDivisorByDivisorGpuTester.DivisorScanSession)
|
||||||
.GetField("_hostBuffer", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
.GetField("_hostBuffer", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
||||||
@@ -240,22 +256,22 @@ public class MersenneNumberDivisorGpuTesterTests
|
|||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_431_as_composite()
|
public void ByDivisor_session_marks_mersenne_numbers_divisible_by_large_divisor2_as_composite()
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorGpuTester
|
var tester = new MersenneNumberDivisorByDivisorGpuTester
|
||||||
{
|
{
|
||||||
GpuBatchSize = 5,
|
GpuBatchSize = 5,
|
||||||
};
|
};
|
||||||
tester.ConfigureFromMaxPrime(193UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] exponents = { 67UL, 71UL, 86UL, 134UL, 172UL };
|
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(4_000_561UL);
|
||||||
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(4_000_561UL, divisorData);
|
||||||
|
|
||||||
|
ulong[] exponents = { 61UL, 73UL, cycle, cycle * 2UL, cycle - 1UL };
|
||||||
byte[] hits = new byte[exponents.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(431UL);
|
session.CheckDivisor(4_000_561UL, divisorData, cycle, exponents, hits);
|
||||||
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(431UL, divisorData);
|
|
||||||
|
|
||||||
session.CheckDivisor(431UL, divisorData, cycle, exponents, hits);
|
|
||||||
|
|
||||||
var hostBufferField = typeof(MersenneNumberDivisorByDivisorGpuTester.DivisorScanSession)
|
var hostBufferField = typeof(MersenneNumberDivisorByDivisorGpuTester.DivisorScanSession)
|
||||||
.GetField("_hostBuffer", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
.GetField("_hostBuffer", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
||||||
@@ -279,44 +295,27 @@ public class MersenneNumberDivisorGpuTesterTests
|
|||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
[Trait("Category", "Fast")]
|
[Trait("Category", "Fast")]
|
||||||
public void ByDivisor_gpu_tester_ignores_small_cycle_overrides()
|
public void ByDivisor_gpu_tester_generates_cycles_for_large_divisors()
|
||||||
{
|
|
||||||
var cycles = MersenneDivisorCycles.Shared;
|
|
||||||
var tableField = typeof(MersenneDivisorCycles).GetField("_table", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
|
||||||
var smallCyclesField = typeof(MersenneDivisorCycles).GetField("_smallCycles", BindingFlags.NonPublic | BindingFlags.Instance)!;
|
|
||||||
|
|
||||||
var originalTable = (List<(ulong Divisor, ulong Cycle)>)tableField.GetValue(cycles)!;
|
|
||||||
var originalSmall = (ulong[]?)smallCyclesField.GetValue(cycles);
|
|
||||||
|
|
||||||
try
|
|
||||||
{
|
{
|
||||||
var tester = new MersenneNumberDivisorByDivisorGpuTester();
|
var tester = new MersenneNumberDivisorByDivisorGpuTester();
|
||||||
tester.ConfigureFromMaxPrime(19UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
tester.IsPrime(19UL, out bool baselineExhausted).Should().BeTrue();
|
using var session = tester.CreateDivisorSession();
|
||||||
baselineExhausted.Should().BeTrue();
|
ulong[] exponents = { 59UL, 67UL, 117_652UL };
|
||||||
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
var patchedTable = new List<(ulong Divisor, ulong Cycle)>(originalTable);
|
MontgomeryDivisorData divisorData = MontgomeryDivisorData.FromModulus(4_000_169UL);
|
||||||
ulong[] patchedSmall = new ulong[PerfectNumberConstants.MaxQForDivisorCycles + 1];
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(4_000_169UL, divisorData);
|
||||||
patchedSmall[191] = 5UL;
|
|
||||||
|
|
||||||
tableField.SetValue(cycles, patchedTable);
|
session.CheckDivisor(4_000_169UL, divisorData, cycle, exponents, hits);
|
||||||
smallCyclesField.SetValue(cycles, patchedSmall);
|
|
||||||
DivisorCycleCache.Shared.RefreshSnapshot();
|
|
||||||
|
|
||||||
tester.IsPrime(19UL, out bool patchedExhausted).Should().BeTrue();
|
byte[] expected = new byte[exponents.Length];
|
||||||
patchedExhausted.Should().BeTrue();
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
|
|
||||||
MersenneDivisorCycles.TryMatchSmallCycleForFactoring(191UL, 5UL, out ulong cachedCycle).Should().BeTrue();
|
|
||||||
cachedCycle.Should().Be(5UL);
|
|
||||||
}
|
|
||||||
finally
|
|
||||||
{
|
{
|
||||||
tableField.SetValue(cycles, originalTable);
|
expected[i] = exponents[i].Pow2ModWindowedCpu(divisorData.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
smallCyclesField.SetValue(cycles, originalSmall);
|
|
||||||
DivisorCycleCache.Shared.RefreshSnapshot();
|
|
||||||
GpuContextPool.DisposeAll();
|
|
||||||
}
|
}
|
||||||
|
|
||||||
|
hits.Should().Equal(expected);
|
||||||
}
|
}
|
||||||
|
|
||||||
[Fact]
|
[Fact]
|
||||||
@@ -328,23 +327,23 @@ public class MersenneNumberDivisorGpuTesterTests
|
|||||||
BatchSize = 2,
|
BatchSize = 2,
|
||||||
};
|
};
|
||||||
|
|
||||||
tester.ConfigureFromMaxPrime(173UL);
|
tester.ConfigureFromMaxPrime(197UL);
|
||||||
|
|
||||||
using var session = tester.CreateDivisorSession();
|
using var session = tester.CreateDivisorSession();
|
||||||
ulong[] primes = { 59UL, 61UL, 67UL, 71UL, 73UL };
|
ulong[] exponents = { 59UL, 61UL, 67UL, 71UL, 73UL, 79UL, 117_652UL };
|
||||||
byte[] hits = new byte[primes.Length];
|
byte[] hits = new byte[exponents.Length];
|
||||||
|
|
||||||
MontgomeryDivisorData divisor223 = MontgomeryDivisorData.FromModulus(223UL);
|
MontgomeryDivisorData divisor = MontgomeryDivisorData.FromModulus(4_000_169UL);
|
||||||
ulong cycle223 = MersenneDivisorCycles.CalculateCycleLength(223UL, divisor223);
|
ulong cycle = MersenneDivisorCycles.CalculateCycleLength(4_000_169UL, divisor);
|
||||||
session.CheckDivisor(223UL, divisor223, cycle223, primes, hits);
|
session.CheckDivisor(4_000_169UL, divisor, cycle, exponents, hits);
|
||||||
|
|
||||||
byte[] expected223 = new byte[primes.Length];
|
byte[] expected = new byte[exponents.Length];
|
||||||
for (int i = 0; i < primes.Length; i++)
|
for (int i = 0; i < exponents.Length; i++)
|
||||||
{
|
{
|
||||||
expected223[i] = primes[i].Pow2ModWindowedCpu(divisor223.Modulus) == 1UL ? (byte)1 : (byte)0;
|
expected[i] = exponents[i].Pow2ModWindowedCpu(divisor.Modulus) == 1UL ? (byte)1 : (byte)0;
|
||||||
}
|
}
|
||||||
|
|
||||||
hits.Should().Equal(expected223);
|
hits.Should().Equal(expected);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
Reference in new issue
Block a user