run_benchmark.py

利用C

__author__ = "Ilmar Wilbers ([e-mail])"
__date__ = "2008-06-04"
__copyright__ = "Copyright (C) 2007 Ilmar Wilbers"
__license__  = "GNU LGPL Version 2.1"

# Be careful to first run the benchmark once in order for
# the Python extensions to be created, or else the timings
# will not be correct.

from dolfin import *
from time import time
import sys

def make_form(name, mesh):
    globals()['mesh'] = mesh
    execfile(name + '.form', globals())
    try:
        return a
    except:
        print "No object 'a' to return in file %s.form" %name
        return None

def bench_form(form, mesh, nr_reasm=1):
    totaltime = 0.0
    t0 = time()
    A = assemble(form, mesh)
    totaltime += time() - t0
    for i in range(nr_reasm - 1):
        t0 = time()
        assemble(form, mesh, tensor=A, reset_tensor=False)
        totaltime += time() - t0
    return totaltime/float(nr_reasm)

def make_mesh(name, dim=2):
    if dim == 3:
        N = 32
        mesh = UnitCube(N, N, N)
        return mesh
    else: # Assume 3D
        N = 256
        mesh = UnitSquare(N, N)
        return mesh

def print_table(values, title):
    "Print nicely formatted table"
    m = max([key[0] for key in values]) + 2
    n = max([key[1] for key in values]) + 2

    table = []
    for i in range(m):
        table.append(["" for j in range(n)])

    for i in range(m - 1):
        table[i + 1][0] = str(values[(i, 0)][0]).split(" ")[0]

    for j in range(n - 1):
        table[0][j + 1] = str(values[(0, j)][1]).split(" ")[0]

    for i in range(m - 1):
        for j in range(n - 1):
            table[i + 1][j + 1] = "%.5g" % values[(i, j)][2]

    table[0][0] = title

    column_sizes = [max([len(table[i][j]) for i in range(m)]) for j in
    range(n)]
    row_size = sum(column_sizes) + 3*(len(column_sizes) - 1) + 2

    print ""
    for i in range(m):
        print " " + "-"*row_size
        print "|",
        for j in range(n):
            print table[i][j] + " "*(column_sizes[j] - len(table[i][j])),
            print "|",
        print ""
    print " " + "-"*row_size
    print ""

if __name__ == "__main__":
    try:
        num_reasm = int(sys.argv[1])
    except:
        print 'Usage: %s number_of_repetitions [1]' %sys.argv[0]
        num_reasm = 1

dolfin_set("output destination", "silent")
backends = ["uBLAS", "PETSc", "Epetra", "Assembly"]
# forms consist of name of the form (same as file)
# and a list of arguments to assembler (given to the constructor in cpp)
# as a tuple.
forms = ["Elasticity3D",
         "PoissonP1", 
         "PoissonP2", 
         "PoissonP3", 
         "THStokes2D",
         "NSEMomentum3D",
         "StabStokes2D",
         ]
results = {}
for i in range(len(backends)):
    backend = backends[i]
    dolfin_set("linear algebra backend", backends[i])
    for j in range(len(forms)):
        form = forms[j]
        dim = 2 if not form.find("3D") > -1 else 3
        m = make_mesh(form, dim)
        a = make_form(form, m)
        print "Assembling %s with %s" %(form, backend)
        results[(i, j)] = (backend, form, bench_form(a, m,
                                                     nr_reasm=num_reasm))

print_table(results, "Backend/Form")