~mogurijin/pyeigen/py3k

Viewing changes to test/benchmark/matrixX.py

Committer: Jussi Lepistö
Date: 2010-04-07 07:24:37 UTC
Revision ID: jussi.lepisto@iki.fi-20100407072437-i8769yqa2hbf7rn9

- Enable vectorization (SSE2 instruction set)
- Add 1000x1000 matrix benchmark
- Change NumPy type from array to matrix in benchmarks

files added:
test/benchmark/matrixX.py

files modified:
TODO.txt

setup.py

test/benchmark/matrix4.py

Show diffs side-by-side

added added

removed removed

test/benchmark/matrixX.py

import timeit

number = 10

repeat = 3

test_pyeigen = True

test_numpy = True

m1 = None

m2 = None

def test(stmt, setup):

return timeit.repeat(stmt, setup, repeat=repeat, number=number)

def run_tests():

global m1, m2

# Pyeigen

try:

import pyeigen

except ImportError:

pyeigen = None

if test_pyeigen and pyeigen is not None:

print "PyEigen"

m1 = pyeigen.MatrixXf.random(1000, 1000)

m2 = pyeigen.MatrixXf.random(1000, 1000)

setup = "from __main__ import m1, m2"

print "Add:", test("m1 + m2", setup)

print "Matrix multiply:", test("m1 * m2", setup)

print "Scalar multiply:", test("m1 * 2.0", setup)

# Numpy

try:

import numpy.matlib

except ImportError:

numpy = None

if test_numpy and numpy is not None:

print "NumPy"

m1 = numpy.matlib.rand((1000, 1000)).astype(numpy.float32)

m2 = numpy.matlib.rand((1000, 1000)).astype(numpy.float32)

setup = "from __main__ import m1, m2"

print "Add:", test("m1 + m2", setup)

print "Matrix multiply:", test("m1 * m2", setup)

print "Scalar multiply:", test("m1 * 2.0", setup)

if __name__ == "__main__":

run_tests()

Older »