提交 7d46f831 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Skip profiling tests in DebugMode.

上级 98ccc988
......@@ -38,6 +38,8 @@ class TestD3Viz(unittest.TestCase):
self.check(f)
def test_mlp_profiled(self):
if th.config.mode in ("DebugMode", "DEBUG_MODE"):
raise SkipTest("Can't profile in DebugMode")
m = models.Mlp()
profile = th.compile.profiling.ProfileStats(False)
f = th.function(m.inputs, m.outputs, profile=profile)
......
......@@ -84,6 +84,9 @@ def test_pydotprint_profile():
if not theano.printing.pydot_imported:
raise SkipTest('pydot not available')
if theano.config.mode in ("DebugMode", "DEBUG_MODE"):
raise SkipTest("Can't profile in DebugMode")
A = tensor.matrix()
prof = theano.compile.ProfileStats(atexit_print=False, gpu_checks=False)
f = theano.function([A], A + 1, profile=prof)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论