提交 eb2f4398 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Use TestOptimizationMixin to provide better error messages when the op is not properly optimized.

上级 21581731
...@@ -107,7 +107,7 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu, ...@@ -107,7 +107,7 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu,
_skip = skip _skip = skip
_checks = checks _checks = checks
class Checker(unittest.TestCase): class Checker(unittest.TestCase, utt.TestOptimizationMixin):
op = staticmethod(_op) op = staticmethod(_op)
gpu_op = staticmethod(_gpu_op) gpu_op = staticmethod(_gpu_op)
cases = _cases cases = _cases
...@@ -144,8 +144,7 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu, ...@@ -144,8 +144,7 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu,
exc.args += (err_msg,) exc.args += (err_msg,)
raise raise
assert any(node.op is self.gpu_op self.assertFunctionContains1(f_tst, self.gpu_op)
for node in f_tst.maker.fgraph.toposort()), testname
ref_e = None ref_e = None
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论