Skip to content

Commit d57d82c

Browse files
committed
Modernize and make MKL benchmark more demanding
1 parent b93a4bb commit d57d82c

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

bench/vml_timing2.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@
1313

1414
import numexpr as ne
1515

16-
N = int(2**26)
16+
N = int(2**28)
1717

1818
x = np.linspace(0, 1, N)
1919
y = np.linspace(0, 1, N)
@@ -31,17 +31,17 @@
3131
print("Time for an algebraic expression: %.3f s / %.3f GB/s" % (t1-t0, gbs))
3232

3333
t0 = time()
34-
z = np.sin(x)**2 + np.cos(y)**2
34+
z = np.sin(x)**3.2 + np.cos(y)**3.2
3535
t1 = time()
3636
gbs = working_set_GB / (t1-t0)
3737
print("Time for a transcendental expression: %.3f s / %.3f GB/s" % (t1-t0, gbs))
3838

3939
if ne.use_vml:
4040
ne.set_vml_num_threads(1)
41-
ne.set_num_threads(8)
41+
ne.set_num_threads(16)
4242
print("NumExpr version: %s, Using MKL ver. %s, Num threads: %s" % (ne.__version__, ne.get_vml_version(), ne.nthreads))
4343
else:
44-
ne.set_num_threads(8)
44+
ne.set_num_threads(16)
4545
print("NumExpr version: %s, Not Using MKL, Num threads: %s" % (ne.__version__, ne.nthreads))
4646

4747
t0 = time()
@@ -51,7 +51,7 @@
5151
print("Time for an algebraic expression: %.3f s / %.3f GB/s" % (t1-t0, gbs))
5252

5353
t0 = time()
54-
ne.evaluate('sin(x)**2 + cos(y)**2', out = z)
54+
ne.evaluate('sin(x)**3.2 + cos(y)**3.2', out = z)
5555
t1 = time()
5656
gbs = working_set_GB / (t1-t0)
5757
print("Time for a transcendental expression: %.3f s / %.3f GB/s" % (t1-t0, gbs))

0 commit comments

Comments
 (0)