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
|
package ff
import (
"math/big"
"reflect"
"testing"
)
func Test_mul128(t *testing.T) {
type args struct {
x [2]uint64
y [2]uint64
}
tests := []struct {
name string
args args
wantZ [4]uint64
}{
{"2*1", args{[2]uint64{2, 0}, [2]uint64{1, 0}}, [4]uint64{2, 0, 0, 0}},
{"(2^128 - 1)(2^128 - 1)", args{[2]uint64{0xffffffffffffffff, 0xffffffffffffffff}, [2]uint64{0xffffffffffffffff, 0xffffffffffffffff}}, [4]uint64{0x1, 0, 0xfffffffffffffffe, 0xffffffffffffffff}},
}
for _, tt := range tests {
t.Run(tt.name, func(t *testing.T) {
if gotZ := mul128(tt.args.x, tt.args.y); !reflect.DeepEqual(gotZ, tt.wantZ) {
t.Errorf("mul128() = %v, want %v", gotZ, tt.wantZ)
}
})
}
}
func Test_mul128_random(t *testing.T) {
for i := 0; i < 100; i++ {
x := [2]uint64{randomElement(2 * i)[0], randomElement(2 * i)[1]}
y := [2]uint64{randomElement(2*i + 1)[0], randomElement(2*i + 1)[1]}
got := mul128(x, y)
want := BigIntToElement(new(big.Int).Mul(ElementToBigInt(Element{x[0], x[1], 0, 0}), ElementToBigInt(Element{y[0], y[1], 0, 0})))
if got != want {
t.Errorf("/%d mul128(%v,%v) = %v, want %v", i, x, y, got, want)
t.FailNow()
}
}
}
func Benchmark_mul128(b *testing.B) {
x := [2]uint64{randomElement(1)[0], randomElement(1)[1]}
y := [2]uint64{randomElement(2)[0], randomElement(2)[1]}
b.ResetTimer()
for i := 0; i < b.N; i++ {
mul128(x, y)
}
}
func Benchmark_mul128_BI(b *testing.B) {
x := [4]uint64{randomElement(1)[0], randomElement(1)[1], 0, 0}
y := [4]uint64{randomElement(2)[0], randomElement(2)[1], 0, 0}
A, B := ElementToBigInt(x), ElementToBigInt(y)
C := new(big.Int)
b.ResetTimer()
for i := 0; i < b.N; i++ {
C.Mul(A, B)
}
}
|
