mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 03:51:29 +00:00
485fc538c3
* Initial version of scatter * Add mutex for scatter batch * Provide mutex to worker processes * Embed mputil * Add notes on scatter * Tidy-up * Add mutex test * Simplify scatter for users * Tidy-ups/code coverage * Gazelle update * Add benchmark * Mutex to RWMutex * Add test against internal functions * Bazel fixes * Fix benchmark * Benchmark values to constants * Update shared/mputil/scatter.go Co-Authored-By: Raul Jordan <raul@prysmaticlabs.com> * Update shared/mputil/scatter.go Co-Authored-By: Raul Jordan <raul@prysmaticlabs.com> * Moved struct to top of file * Add featureconfig for scatter * Gate scatter behind feature flag * Lint fixes * fmt
58 lines
1.2 KiB
Go
58 lines
1.2 KiB
Go
package mputil_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/mputil"
|
|
)
|
|
|
|
var input [][]byte
|
|
|
|
const (
|
|
benchmarkElements = 65536
|
|
benchmarkElementSize = 32
|
|
benchmarkHashRuns = 128
|
|
)
|
|
|
|
func init() {
|
|
input = make([][]byte, benchmarkElements)
|
|
for i := 0; i < benchmarkElements; i++ {
|
|
input[i] = make([]byte, benchmarkElementSize)
|
|
rand.Read(input[i])
|
|
}
|
|
}
|
|
|
|
// hash repeatedly hashes the data passed to it
|
|
func hash(input [][]byte) [][]byte {
|
|
output := make([][]byte, len(input))
|
|
for i := range input {
|
|
copy(output, input)
|
|
for j := 0; j < benchmarkHashRuns; j++ {
|
|
hash := sha256.Sum256(output[i])
|
|
output[i] = hash[:]
|
|
}
|
|
}
|
|
return output
|
|
}
|
|
|
|
func BenchmarkHash(b *testing.B) {
|
|
for i := 0; i < b.N; i++ {
|
|
hash(input)
|
|
}
|
|
}
|
|
|
|
func BenchmarkHashMP(b *testing.B) {
|
|
output := make([][]byte, len(input))
|
|
for i := 0; i < b.N; i++ {
|
|
workerResults, _ := mputil.Scatter(len(input), func(offset int, entries int, _ *sync.RWMutex) (interface{}, error) {
|
|
return hash(input[offset : offset+entries]), nil
|
|
})
|
|
for _, result := range workerResults {
|
|
copy(output[result.Offset:], result.Extent.([][]byte))
|
|
}
|
|
}
|
|
}
|