-
Notifications
You must be signed in to change notification settings - Fork 6
/
aggregates_amd64.go
87 lines (80 loc) · 1.55 KB
/
aggregates_amd64.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
package functions
import (
"github.com/viterin/partial"
"golang.org/x/exp/slices"
"math"
)
func Mean_AVX2_F64(x []float64) float64 {
return Sum_AVX2_F64(x) / float64(len(x))
}
func Mean_AVX2_F32(x []float32) float32 {
return Sum_AVX2_F32(x) / float32(len(x))
}
func Median_AVX2_F64(x []float64) float64 {
if len(x)%2 == 1 {
x = slices.Clone(x)
i := len(x) / 2
partial.TopK(x, i+1)
return x[i]
}
return Quantile_AVX2_F64(x, 0.5)
}
func Median_AVX2_F32(x []float32) float32 {
if len(x)%2 == 1 {
x = slices.Clone(x)
i := len(x) / 2
partial.TopK(x, i+1)
return x[i]
}
return Quantile_AVX2_F32(x, 0.5)
}
func Quantile_AVX2_F64(x []float64, q float64) float64 {
if len(x) == 1 {
return x[0]
}
if q == 0 {
return Min_AVX2_F64(x)
}
if q == 1 {
return Max_AVX2_F64(x)
}
x = slices.Clone(x)
f := float64(len(x)-1) * q
i := int(math.Floor(f))
if q < 0.5 {
partial.TopK(x, i+2)
a := Max_AVX2_F64(x[:i+1])
b := x[i+1]
return a + (b-a)*(f-float64(i))
} else {
partial.TopK(x, i+1)
a := x[i]
b := Min_AVX2_F64(x[i+1:])
return a + (b-a)*(f-float64(i))
}
}
func Quantile_AVX2_F32(x []float32, q float32) float32 {
if len(x) == 1 {
return x[0]
}
if q == 0 {
return Min_AVX2_F32(x)
}
if q == 1 {
return Max_AVX2_F32(x)
}
x = slices.Clone(x)
f := float32(len(x)-1) * q
i := int(math.Floor(float64(f)))
if q < 0.5 {
partial.TopK(x, i+2)
a := Max_AVX2_F32(x[:i+1])
b := x[i+1]
return a + (b-a)*(f-float32(i))
} else {
partial.TopK(x, i+1)
a := x[i]
b := Min_AVX2_F32(x[i+1:])
return a + (b-a)*(f-float32(i))
}
}