提交 9093449e authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Bypass GPU checks in tests

The gpu back-end may have been imported, but we are not even timing a GPU function
上级 9f04330b
......@@ -35,7 +35,7 @@ class Test_profiling(unittest.TestCase):
z += [T.outer(x[i], x[i + 1]).sum(axis=1) for i in range(len(x) - 1)]
z += [x[i] + x[i + 1] for i in range(len(x) - 1)]
p = theano.ProfileStats(False)
p = theano.ProfileStats(False, gpu_checks=False)
if theano.config.mode in ["DebugMode", "DEBUG_MODE", "FAST_COMPILE"]:
m = "FAST_RUN"
......@@ -87,7 +87,7 @@ class Test_profiling(unittest.TestCase):
z = ifelse(T.lt(a, b), x * 2, y * 2)
p = theano.ProfileStats(False)
p = theano.ProfileStats(False, gpu_checks=False)
if theano.config.mode in ["DebugMode", "DEBUG_MODE", "FAST_COMPILE"]:
m = "FAST_RUN"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论