-
Notifications
You must be signed in to change notification settings - Fork 135
/
arith_amd64_test.go
141 lines (118 loc) · 3.99 KB
/
arith_amd64_test.go
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
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
// Code generated by go generate; DO NOT EDIT.
// This file was generated by robots.
//go:build amd64 && !purego
// +build amd64,!purego
package p751
import (
"reflect"
"testing"
"testing/quick"
"github.com/cloudflare/circl/dh/sidh/internal/common"
"golang.org/x/sys/cpu"
)
type OptimFlag uint
const (
// Indicates that optimisation which uses MUL instruction should be used
kUse_MUL OptimFlag = 1 << 0
// Indicates that optimisation which uses MULX instruction should be used
kUse_MULX = 1 << 1
// Indicates that optimisation which uses MULX, ADOX and ADCX instructions should be used
kUse_MULXandADxX = 1 << 2
)
func resetCpuFeatures() {
HasBMI2 = cpu.X86.HasBMI2
HasADXandBMI2 = cpu.X86.HasBMI2 && cpu.X86.HasADX
}
// Utility function used for testing Mul implementations. Tests caller provided
// mulFunc against mul()
func testMul(t *testing.T, f1, f2 OptimFlag) {
doMulTest := func(multiplier, multiplicant common.Fp) bool {
defer resetCpuFeatures()
var resMulRef, resMulOptim common.FpX2
// Compute multiplier*multiplicant with first implementation
HasBMI2 = (kUse_MULX & f1) == kUse_MULX
HasADXandBMI2 = (kUse_MULXandADxX & f1) == kUse_MULXandADxX
mulP751(&resMulOptim, &multiplier, &multiplicant)
// Compute multiplier*multiplicant with second implementation
HasBMI2 = (kUse_MULX & f2) == kUse_MULX
HasADXandBMI2 = (kUse_MULXandADxX & f2) == kUse_MULXandADxX
mulP751(&resMulRef, &multiplier, &multiplicant)
// Compare results
return reflect.DeepEqual(resMulRef, resMulOptim)
}
if err := quick.Check(doMulTest, quickCheckConfig); err != nil {
t.Error(err)
}
}
// Utility function used for testing REDC implementations. Tests caller provided
// redcFunc against redc()
func testRedc(t *testing.T, f1, f2 OptimFlag) {
doRedcTest := func(aRR common.FpX2) bool {
defer resetCpuFeatures()
var resRedcF1, resRedcF2 common.Fp
aRRcpy := aRR
// Compute redc with first implementation
HasBMI2 = (kUse_MULX & f1) == kUse_MULX
HasADXandBMI2 = (kUse_MULXandADxX & f1) == kUse_MULXandADxX
rdcP751(&resRedcF1, &aRR)
// Compute redc with second implementation
HasBMI2 = (kUse_MULX & f2) == kUse_MULX
HasADXandBMI2 = (kUse_MULXandADxX & f2) == kUse_MULXandADxX
rdcP751(&resRedcF2, &aRRcpy)
// Compare results
return reflect.DeepEqual(resRedcF2, resRedcF1)
}
if err := quick.Check(doRedcTest, quickCheckConfig); err != nil {
t.Error(err)
}
}
// Ensures correctness of implementation of mul operation which uses MULX
func TestMulWithMULX(t *testing.T) {
defer resetCpuFeatures()
if !HasBMI2 {
t.Skip("MULX not supported by the platform")
}
testMul(t, kUse_MULX, kUse_MUL)
}
// Ensures correctness of implementation of mul operation which uses MULX and ADOX/ADCX
func TestMulWithMULXADxX(t *testing.T) {
defer resetCpuFeatures()
if !HasADXandBMI2 {
t.Skip("MULX, ADCX and ADOX not supported by the platform")
}
testMul(t, kUse_MULXandADxX, kUse_MUL)
}
// Ensures correctness of implementation of mul operation which uses MULX and ADOX/ADCX
func TestMulWithMULXADxXAgainstMULX(t *testing.T) {
defer resetCpuFeatures()
if !HasADXandBMI2 {
t.Skip("MULX, ADCX and ADOX not supported by the platform")
}
testMul(t, kUse_MULX, kUse_MULXandADxX)
}
// Ensures correctness of Montgomery reduction implementation which uses MULX
func TestRedcWithMULX(t *testing.T) {
defer resetCpuFeatures()
if !HasBMI2 {
t.Skip("MULX not supported by the platform")
}
testRedc(t, kUse_MULX, kUse_MUL)
}
// Ensures correctness of Montgomery reduction implementation which uses MULX
// and ADCX/ADOX.
func TestRedcWithMULXADxX(t *testing.T) {
defer resetCpuFeatures()
if !HasADXandBMI2 {
t.Skip("MULX, ADCX and ADOX not supported by the platform")
}
testRedc(t, kUse_MULXandADxX, kUse_MUL)
}
// Ensures correctness of Montgomery reduction implementation which uses MULX
// and ADCX/ADOX.
func TestRedcWithMULXADxXAgainstMULX(t *testing.T) {
defer resetCpuFeatures()
if !HasADXandBMI2 {
t.Skip("MULX, ADCX and ADOX not supported by the platform")
}
testRedc(t, kUse_MULXandADxX, kUse_MULX)
}