Refine Montgomery GPU helpers to avoid kernel branching

This commit is contained in:
Marek Lemańczyk committed 2025-10-22 01:43:10 +02:00
1 parent f12919e220
commit 24eaec7a21
5 files changed
+46 -22

No files matched your search

@@ -12,7 +12,7 @@ internal static class Pow2MontgomeryKernels
MontgomeryDivisorData divisor,
ArrayView1D<ulong, Stride1D.Dense> results)
{
results[index] = ULongExtensions.Pow2MontgomeryModWindowedGpu(divisor, exponents[index], keepMontgomery: true);
results[index] = ULongExtensions.Pow2MontgomeryModWindowedGpuKeepMontgomery(divisor, exponents[index]);
}
public static void Pow2MontgomeryKernelConvertToStandard(
@@ -21,6 +21,6 @@ internal static class Pow2MontgomeryKernels
MontgomeryDivisorData divisor,
ArrayView1D<ulong, Stride1D.Dense> results)
{
results[index] = ULongExtensions.Pow2MontgomeryModWindowedGpu(divisor, exponents[index], keepMontgomery: false);
results[index] = ULongExtensions.Pow2MontgomeryModWindowedGpuConvertToStandard(divisor, exponents[index]);
}
}