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

fix test with gpu to the new interface.

上级 35d29d38
......@@ -30,6 +30,8 @@ def get_mode(use_gpu):
ret = theano.compile.ProfileMode()
if use_gpu:
ret = ret.including('gpu')
else:
ret = ret.excluding('gpu')
return ret
def print_mode(mode):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论