Author: Brian Kearns <bdkea...@gmail.com>
Branch: 
Changeset: r69304:74eff28ac998
Date: 2014-02-23 16:57 -0500
http://bitbucket.org/pypy/pypy/changeset/74eff28ac998/

Log:    add dot bench script

diff --git a/pypy/module/micronumpy/bench/dot.py 
b/pypy/module/micronumpy/bench/dot.py
new file mode 100644
--- /dev/null
+++ b/pypy/module/micronumpy/bench/dot.py
@@ -0,0 +1,28 @@
+import time
+
+try:
+    import numpypy
+except ImportError:
+    pass
+
+import numpy
+
+def get_matrix():
+    import random
+    n = 502
+    x = numpy.zeros((n,n), dtype=numpy.float64)
+    for i in range(n):
+        for j in range(n):
+            x[i][j] = random.random()
+    return x
+
+def main():
+    x = get_matrix()
+    y = get_matrix()
+    a = time.time()
+    #z = numpy.dot(x, y)  # uses numpy possibly-blas-lib dot
+    z = numpy.core.multiarray.dot(x, y)  # uses strictly numpy C dot
+    b = time.time()
+    print '%.2f seconds' % (b-a)
+
+main()
_______________________________________________
pypy-commit mailing list
pypy-commit@python.org
https://mail.python.org/mailman/listinfo/pypy-commit

Reply via email to