mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2024-12-23 11:57:18 +00:00
f61f02e59b
* cmd tests * deposit util tests * feature config tests * hashutil tests * htr util tests * interop tests * ip util tests * Update BLS * Update cmd * Update bytesutil and depositutil * Update event * Update keystore * Update mathutil * Update mputil * Update pagination * Update params * Upate prome * Update testutil * Update trieutil * Merge branch 'master' of github.com:prysmaticlabs/prysm into testutil-shared * Sync with master * Mod * Typo * Revert * gazelle * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Gaz * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * Merge refs/heads/master into testutil-shared * fixes build error * Merge refs/heads/master into testutil-shared
64 lines
1.4 KiB
Go
64 lines
1.4 KiB
Go
package mputil_test
|
|
|
|
import (
|
|
"crypto/rand"
|
|
"crypto/sha256"
|
|
"sync"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/mputil"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
log "github.com/sirupsen/logrus"
|
|
)
|
|
|
|
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)
|
|
_, err := rand.Read(input[i])
|
|
if err != nil {
|
|
log.WithError(err).Debug("Cannot read from rand")
|
|
}
|
|
}
|
|
}
|
|
|
|
// 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, err := mputil.Scatter(len(input), func(offset int, entries int, _ *sync.RWMutex) (interface{}, error) {
|
|
return hash(input[offset : offset+entries]), nil
|
|
})
|
|
require.NoError(b, err)
|
|
for _, result := range workerResults {
|
|
copy(output[result.Offset:], result.Extent.([][]byte))
|
|
}
|
|
}
|
|
}
|