Created
September 14, 2017 10:12
-
-
Save guoxiaolu/b16c5eac627d9d17d153484b51401a60 to your computer and use it in GitHub Desktop.
testGPU
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
from theano import function, config, shared, tensor | |
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([], tensor.exp(x)) | |
print(f.maker.fgraph.toposort()) | |
t0 = time.time() | |
for i in range(iters): | |
r = f() | |
t1 = time.time() | |
print("Looping %d times took %f seconds" % (iters, t1 - t0)) | |
print("Result is %s" % (r,)) | |
if numpy.any([isinstance(x.op, tensor.Elemwise) and | |
('Gpu' not in type(x.op).__name__) | |
for x in f.maker.fgraph.toposort()]): | |
print('Used the cpu') | |
else: | |
print('Used the gpu') |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment