import distarray as da
def f(comm, size, reps, dtype):
"""Benchmark da.ones"""
for i in range(reps):
a = da.ones((size, size), dtype=dtype, comm=comm)
for size, reps in zip([1000, 2000, 4000], 3 * [10]):
sizes, times = da.benchmark_function(f, size, reps, 'float64')
if da.COMM_PRIVATE.Get_rank() == 0:
print
print "array_size, reps:", size, reps
print sizes
print times
import distarray as da
def f(comm, size, reps):
a = da.random.rand((size,size),comm=comm)
b = da.random.rand((size,size),comm=comm)
for i in range(reps):
c = da.cos(10.0*a) + da.sin(b/20.0)
for size, reps in zip([1000,2000,4000],3*[10]):
sizes, times, speedups = da.benchmark_function(f, size, reps)
if da.COMM_PRIVATE.Get_rank()==0:
print
print "array_size, reps:", size, reps
print sizes
print times
print speedups
import distarray as da
def f(comm, size, reps):
"""Benchmark da.random.rand"""
for i in range(reps):
a = da.random.rand((size,size), comm=comm)
for size, reps in zip([1000,2000,4000],3*[10]):
sizes, times = da.benchmark_function(f, size, reps)
if da.COMM_PRIVATE.Get_rank()==0:
print
print "array_size, reps:", size, reps
print sizes
print times
import distarray as da
def f(comm, size, reps, dtype):
"""Benchmark da.empty"""
for i in range(reps):
a = da.empty((size,size), dtype=dtype, comm=comm)
for size, reps in zip([1000,2000,4000],3*[10]):
sizes, times = da.benchmark_function(f, size, reps, 'float64')
if da.COMM_PRIVATE.Get_rank()==0:
print
print "array_size, reps:", size, reps
print sizes
print times