Testing BLAS extension for Bohrium

ext-blas.py

a = np.random.rand(2000, 3000).astype(np.float32)
b = np.random.rand(3000, 4000).astype(np.float32)

t0 = time()
c = np.dot(a, b)
t1 = time()
print "[Numpy]       It took {}s".format(t1-t0)
print c
#print

a = bh.array(a)
b = bh.array(b)
bh.flush()

bh.flush()
t0 = time()
c = bh.blas.gemm(a, b)
bh.flush()
t1 = time()
print "[Bohrium EXT] It took {}s".format(t1-t0)
print c
#print

bh.flush()
t0 = time()
c = bh.dot(a, b)
bh.flush()
t1 = time()
print "[Bohrium DOT] It took {}s".format(t1-t0)
#print c
#print