From 9a796de23083a6342a24f8280c5fe22a2b7b24e3 Mon Sep 17 00:00:00 2001 From: Chunk Date: Mon, 20 Apr 2015 14:30:26 +0800 Subject: [PATCH] staged. --- mmodel/theano/example/gpu_availabe.py | 24 ++++++++++++++++++++++++ 1 file changed, 24 insertions(+), 0 deletions(-) create mode 100644 mmodel/theano/example/gpu_availabe.py diff --git a/mmodel/theano/example/gpu_availabe.py b/mmodel/theano/example/gpu_availabe.py new file mode 100644 index 0000000..9178a82 --- /dev/null +++ b/mmodel/theano/example/gpu_availabe.py @@ -0,0 +1,24 @@ +__author__ = 'chunk' + +from theano import function, config, shared, sandbox +import theano.tensor as T +import numpy +import time + +vlen = 10 * 30 * 768 # 10 x #cores x # threads per core +iters = 1000 + +rng = numpy.random.RandomState(22) +x = shared(numpy.asarray(rng.rand(vlen), config.floatX)) +f = function([], T.exp(x)) +print f.maker.fgraph.toposort() +t0 = time.time() +for i in xrange(iters): + r = f() +t1 = time.time() +print 'Looping %d times took' % iters, t1 - t0, 'seconds' +print 'Result is', r +if numpy.any([isinstance(x.op, T.Elemwise) for x in f.maker.fgraph.toposort()]): + print 'Used the cpu' +else: + print 'Used the gpu' -- libgit2 0.21.2