BENCH: use initialized memory for count_nonzero benchmark

juliantaylor · juliantaylor · commit 6a4858cc1b41 · 2017-03-25T14:38:24.000+01:00
For the bool case uninitialized memory can randomly go into the slow
path handling values different than 0 or 1.
diff --git a/benchmarks/benchmarks/bench_core.py b/benchmarks/benchmarks/bench_core.py
@@ -137,8 +137,8 @@ class CountNonzero(Benchmark):
     ]
 
     def setup(self, numaxes, size, dtype):
-        self.x = np.empty(shape=(
-            numaxes, size), dtype=dtype)
+        self.x = np.arange(numaxes * size).reshape(numaxes, size)
+        self.x = (self.x % 3).astype(dtype)
 
     def time_count_nonzero(self, numaxes, size, dtype):
         np.count_nonzero(self.x)

Original file line number	Diff line number	Diff line change
`@@ -137,8 +137,8 @@ class CountNonzero(Benchmark):`
`137`	`137`	`]`
`138`	`138`
`139`	`139`	`def setup(self, numaxes, size, dtype):`
`140`		`- self.x = np.empty(shape=(`
`141`		`- numaxes, size), dtype=dtype)`
	`140`	`+ self.x = np.arange(numaxes * size).reshape(numaxes, size)`
	`141`	`+ self.x = (self.x % 3).astype(dtype)`
`142`	`142`
`143`	`143`	`def time_count_nonzero(self, numaxes, size, dtype):`
`144`	`144`	`np.count_nonzero(self.x)`