mirror of
https://gitlab.com/pulsechaincom/prysm-pulse.git
synced 2025-01-10 19:51:20 +00:00
5aac06f04e
* begin move * use same import path * imports * regen protos * regen * no rename * generate ssz * gaz * fmt * edit build file * imports * modify * remove generated files * remove protos * edit imports in prysm * beacon chain all builds * edit script * add generated pbs * add replace rules * license for ethereumapis protos * change visibility * fmt * update build files to gaz ignore * use proper form * edit imports * wrap block * revert scripts * revert go mod
105 lines
3.1 KiB
Go
105 lines
3.1 KiB
Go
package attestations
|
|
|
|
import (
|
|
"fmt"
|
|
"testing"
|
|
|
|
"github.com/prysmaticlabs/prysm/shared/copyutil"
|
|
|
|
"github.com/prysmaticlabs/go-bitfield"
|
|
ethpb "github.com/prysmaticlabs/prysm/proto/eth/v1alpha1"
|
|
aggtesting "github.com/prysmaticlabs/prysm/shared/aggregation/testing"
|
|
"github.com/prysmaticlabs/prysm/shared/bls"
|
|
"github.com/prysmaticlabs/prysm/shared/bls/common"
|
|
"github.com/prysmaticlabs/prysm/shared/featureconfig"
|
|
"github.com/prysmaticlabs/prysm/shared/params"
|
|
"github.com/prysmaticlabs/prysm/shared/testutil/require"
|
|
)
|
|
|
|
func BenchmarkAggregateAttestations_Aggregate(b *testing.B) {
|
|
// Override expensive BLS aggregation method with cheap no-op such that this benchmark profiles
|
|
// the logic of aggregation selection rather than BLS logic.
|
|
aggregateSignatures = func(sigs []common.Signature) common.Signature {
|
|
return sigs[0]
|
|
}
|
|
signatureFromBytes = func(sig []byte) (common.Signature, error) {
|
|
return bls.NewAggregateSignature(), nil
|
|
}
|
|
defer func() {
|
|
aggregateSignatures = bls.AggregateSignatures
|
|
signatureFromBytes = bls.SignatureFromBytes
|
|
}()
|
|
|
|
bitlistLen := params.BeaconConfig().MaxValidatorsPerCommittee
|
|
|
|
tests := []struct {
|
|
name string
|
|
inputs []bitfield.Bitlist
|
|
}{
|
|
{
|
|
name: "256 attestations with single bit set",
|
|
inputs: aggtesting.BitlistsWithSingleBitSet(256, bitlistLen),
|
|
},
|
|
{
|
|
name: "256 attestations with 64 random bits set",
|
|
inputs: aggtesting.BitlistsWithSingleBitSet(256, bitlistLen),
|
|
},
|
|
{
|
|
name: "512 attestations with single bit set",
|
|
inputs: aggtesting.BitlistsWithSingleBitSet(512, bitlistLen),
|
|
},
|
|
{
|
|
name: "1024 attestations with 64 random bits set",
|
|
inputs: aggtesting.BitlistsWithMultipleBitSet(b, 1024, bitlistLen, 64),
|
|
},
|
|
}
|
|
|
|
runner := func(atts []*ethpb.Attestation) {
|
|
attsCopy := make([]*ethpb.Attestation, len(atts))
|
|
for i, att := range atts {
|
|
attsCopy[i] = copyutil.CopyAttestation(att)
|
|
}
|
|
_, err := Aggregate(attsCopy)
|
|
require.NoError(b, err)
|
|
}
|
|
|
|
for _, tt := range tests {
|
|
b.Run(fmt.Sprintf("naive_%s", tt.name), func(b *testing.B) {
|
|
b.StopTimer()
|
|
resetCfg := featureconfig.InitWithReset(&featureconfig.Flags{
|
|
AttestationAggregationStrategy: string(NaiveAggregation),
|
|
})
|
|
atts := aggtesting.MakeAttestationsFromBitlists(tt.inputs)
|
|
defer resetCfg()
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
runner(atts)
|
|
}
|
|
})
|
|
b.Run(fmt.Sprintf("max-cover_%s", tt.name), func(b *testing.B) {
|
|
b.StopTimer()
|
|
resetCfg := featureconfig.InitWithReset(&featureconfig.Flags{
|
|
AttestationAggregationStrategy: string(MaxCoverAggregation),
|
|
})
|
|
atts := aggtesting.MakeAttestationsFromBitlists(tt.inputs)
|
|
defer resetCfg()
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
runner(atts)
|
|
}
|
|
})
|
|
b.Run(fmt.Sprintf("opt-max-cover_%s", tt.name), func(b *testing.B) {
|
|
b.StopTimer()
|
|
resetCfg := featureconfig.InitWithReset(&featureconfig.Flags{
|
|
AttestationAggregationStrategy: string(OptMaxCoverAggregation),
|
|
})
|
|
atts := aggtesting.MakeAttestationsFromBitlists(tt.inputs)
|
|
defer resetCfg()
|
|
b.StartTimer()
|
|
for i := 0; i < b.N; i++ {
|
|
runner(atts)
|
|
}
|
|
})
|
|
}
|
|
}
|