提交 e131aa3d authored 作者: Frederic Bastien's avatar Frederic Bastien

added a first version of a script to check that blas work and what version is used.

上级 9c9b7e0b
import theano,numpy,time
import theano.tensor as T
shapes=(2000,2000)
iters = 10
a=T.matrix()
b=T.matrix()
c=theano.shared(numpy.ones(shapes))
f=theano.function([a,b],updates={c:0.4*c+.8*T.dot(a,b)})
print 'blas.ldflags=',theano.config.blas.ldflags
print 'compiledir=',theano.config.compiledir
print f.maker.env.toposort()
av=numpy.ones(shapes)
bv=numpy.ones(shapes)
t0=time.time()
for i in range(iters):
f(av,bv)
print 'times=%.3fs'%(time.time()-t0)
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论