1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
use std::any::TypeId;
#[cfg(all(feature = "cuda", target_arch = "x86_64"))]
use std::sync::atomic::{AtomicBool, Ordering};
use snarkvm_curves::{bls12_377::G1Affine, traits::AffineCurve};
use snarkvm_fields::{PrimeField, Zero};
use snarkvm_utilities::BitIteratorBE;
mod standard;
#[cfg(all(feature = "cuda", target_arch = "x86_64"))]
mod cuda;
#[cfg(all(feature = "cuda", target_arch = "x86_64"))]
static HAS_CUDA_FAILED: AtomicBool = AtomicBool::new(false);
pub struct VariableBaseMSM;
impl VariableBaseMSM {
#[allow(unused)]
fn msm_naive<G: AffineCurve>(bases: &[G], scalars: &[<G::ScalarField as PrimeField>::BigInteger]) -> G::Projective {
let mut acc = G::Projective::zero();
for (base, scalar) in bases.iter().zip(scalars.iter()) {
acc += base.mul_bits(BitIteratorBE::new(*scalar));
}
acc
}
pub fn multi_scalar_mul<G: AffineCurve>(
bases: &[G],
scalars: &[<G::ScalarField as PrimeField>::BigInteger],
) -> G::Projective {
if TypeId::of::<G>() == TypeId::of::<G1Affine>() {
#[cfg(all(feature = "cuda", target_arch = "x86_64"))]
{
if !HAS_CUDA_FAILED.load(Ordering::SeqCst) {
match cuda::msm_cuda(bases, scalars) {
Ok(x) => return x,
Err(e) => {
HAS_CUDA_FAILED.store(true, Ordering::SeqCst);
eprintln!("CUDA failed, moving to next msm method: {:?}", e);
}
}
}
}
}
standard::msm_standard(bases, scalars)
}
}
#[cfg(test)]
mod tests {
use super::*;
use rand::SeedableRng;
use rand_xorshift::XorShiftRng;
use snarkvm_curves::{
bls12_377::{Fr, G1Affine, G1Projective},
traits::ProjectiveCurve,
};
use snarkvm_fields::PrimeField;
use snarkvm_utilities::{rand::UniformRand, BigInteger256};
fn test_data(seed: u64, samples: usize) -> (Vec<G1Affine>, Vec<BigInteger256>) {
let mut rng = XorShiftRng::seed_from_u64(seed);
let v = (0..samples).map(|_| Fr::rand(&mut rng).to_repr()).collect::<Vec<_>>();
let g = (0..samples)
.map(|_| G1Projective::rand(&mut rng).into_affine())
.collect::<Vec<_>>();
(g, v)
}
#[test]
fn test_naive() {
let (bases, scalars) = test_data(334563456, 100);
let rust = standard::msm_standard(bases.as_slice(), scalars.as_slice());
let naive = VariableBaseMSM::msm_naive(bases.as_slice(), scalars.as_slice());
assert_eq!(rust, naive);
}
#[cfg(all(feature = "cuda", target_arch = "x86_64"))]
#[test]
fn test_msm_cuda() {
for i in 0..100 {
let (bases, scalars) = test_data(334563456 + i as u64, 1 << 10);
let rust = standard::msm_standard(bases.as_slice(), scalars.as_slice());
let cuda = cuda::msm_cuda(bases.as_slice(), scalars.as_slice()).unwrap();
assert_eq!(rust, cuda);
}
}
}