prog: measure memory allocation in benchmarks

Enable ReportAllocs.
Also factor out common code into a helper function
(duplicated in 3 places now).

1 files changed, 11 insertions, 0 deletions

diff --git a/prog/export_test.go b/prog/export_test.go
index 1aaea30dc..9ea3f612e 100644
--- a/prog/export_test.go
+++ b/prog/export_test.go
@@ -89,3 +89,14 @@ func testEachTargetRandom(t *testing.T, fn func(t *testing.T, target *Target, rs
 		})
 	}
 }
+
+func initBench(b *testing.B) (*Target, func()) {
+	olddebug := debug
+	debug = false
+	target, err := GetTarget("linux", "amd64")
+	if err != nil {
+		b.Fatal(err)
+	}
+	b.ReportAllocs()
+	return target, func() { debug = olddebug }
+}