from __future__ import print_function
import distarray as da
from benchmark import benchmark_function
def f(comm, size, reps):
"""Benchmark da.add"""
a = da.random.rand((size,size), comm=comm)
b = da.random.rand((size,size), comm=comm)
c = da.empty_like(a)
for i in range(reps):
da.add(a,b,c)
for size, reps in zip([1000,2000,4000],3*[10]):
sizes, times, speedups = benchmark_function(f, size, reps)
if da.mpiutils.COMM_PRIVATE.Get_rank()==0:
print()
print("array_size, reps:", size, reps)
print(sizes)
print(times)
print(speedups)
from __future__ import print_function
import distarray as da
from benchmark import benchmark_function
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, speedups = benchmark_function(f, size, reps, 'float64')
if da.mpiutils.COMM_PRIVATE.Get_rank()==0:
print()
print("array_size, reps:", size, reps)
print(sizes)
print(times)