提交 cc2b3a66 authored 作者: Frederic's avatar Frederic

Fix the detection of cpu vs gpu in the tutorial.

new op wasn't added there.
上级 fb77d434
...@@ -57,11 +57,11 @@ The following output depicts the pre- and post- compilation graphs. ...@@ -57,11 +57,11 @@ The following output depicts the pre- and post- compilation graphs.
predict = theano.function(inputs=[x], outputs=prediction, predict = theano.function(inputs=[x], outputs=prediction,
name="predict") name="predict")
if any( [x.op.__class__.__name__=='Gemv' for x in if any([x.op.__class__.__name__ in ['Gemv', 'CGemv'] for x in
train.maker.fgraph.toposort()]): train.maker.fgraph.toposort()]):
print 'Used the cpu' print 'Used the cpu'
elif any( [x.op.__class__.__name__=='GpuGemm' for x in elif any([x.op.__class__.__name__ == 'GpuGemm' for x in
train.maker.fgraph.toposort()]): train.maker.fgraph.toposort()]):
print 'Used the gpu' print 'Used the gpu'
else: else:
print 'ERROR, not able to tell if theano used the cpu or the gpu' print 'ERROR, not able to tell if theano used the cpu or the gpu'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论