Skip to content

Commit 153e5f8

Browse files
committed
Use larger arrays for benchs
1 parent 619b122 commit 153e5f8

File tree

3 files changed

+7
-6
lines changed

3 files changed

+7
-6
lines changed

bench/boolean_timing.py

+5-4
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313
import timeit
1414
import numpy
1515

16-
array_size = 1000*1000
16+
array_size = 5_000_000
1717
iterations = 10
1818

1919
numpy_ttime = []
@@ -23,6 +23,7 @@
2323
numexpr_sttime = []
2424
numexpr_nttime = []
2525

26+
2627
def compare_times(expr, nexpr):
2728
global numpy_ttime
2829
global numpy_sttime
@@ -64,15 +65,15 @@ def compare_times(expr, nexpr):
6465
numexpr_stime = round(numexpr_timer.timeit(number=iterations), 4)
6566
numexpr_sttime.append(numexpr_stime)
6667
print("numexpr strided:", numexpr_stime/iterations, end=" ")
67-
print("Speed-up of numexpr strided over numpy:", \
68+
print("Speed-up of numexpr strided over numpy:",
6869
round(numpy_stime/numexpr_stime, 4))
6970

7071
evalexpr = 'evaluate("%s", optimization="aggressive")' % expr
7172
numexpr_timer = timeit.Timer(evalexpr, setup_unaligned)
7273
numexpr_ntime = round(numexpr_timer.timeit(number=iterations), 4)
7374
numexpr_nttime.append(numexpr_ntime)
7475
print("numexpr unaligned:", numexpr_ntime/iterations, end=" ")
75-
print("Speed-up of numexpr unaligned over numpy:", \
76+
print("Speed-up of numexpr unaligned over numpy:",
7677
round(numpy_ntime/numexpr_ntime, 4))
7778

7879

@@ -113,7 +114,7 @@ def compare_times(expr, nexpr):
113114
expressions.append('sqrt(i2**2 + f3**2) > 1')
114115
expressions.append('(i2>2) | ((f3**2>3) & ~(i2*f3<2))')
115116

116-
def compare(expression=False):
117+
def compare(expression=None):
117118
if expression:
118119
compare_times(expression, 1)
119120
sys.exit(0)

bench/timing.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,7 @@
1111
from __future__ import print_function
1212
import timeit, numpy
1313

14-
array_size = 1e6
14+
array_size = 5e6
1515
iterations = 2
1616

1717
# Choose the type you want to benchmark

bench/vml_timing.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@
1414
import numpy
1515
import numexpr
1616

17-
array_size = 1000*1000
17+
array_size = 5_000_000
1818
iterations = 10
1919

2020
numpy_ttime = []

0 commit comments

Comments
 (0)