Collect memory stats during AEAD benchmarks.

PiperOrigin-RevId: 614563652
Change-Id: Ic84d75c34efc724b1f8a4a62bebe3b1d62f9e66e

aead/aead_benchmark_test.go

@@ -27,11 +27,13 @@ import (
 
 // Benchmarks for AEAD algorithms.
 
-const plaintextSize = 16 * 1024
-const associatedDataSize = 256
-
 func BenchmarkEncryptDecrypt(b *testing.B) {
-	var testCases = []struct {
+	const (
+		plaintextSize      = 16 * 1024
+		associatedDataSize = 256
+	)
+
+	testCases := []struct {
 		name     string
 		template *tinkpb.KeyTemplate
 	}{
@@ -63,6 +65,8 @@ func BenchmarkEncryptDecrypt(b *testing.B) {
 	}
 	for _, tc := range testCases {
 		b.Run(tc.name, func(b *testing.B) {
+			b.ReportAllocs()
+
 			handle, err := keyset.NewHandle(tc.template)
 			if err != nil {
 				b.Fatal(err)