提交 8cc4bb52 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Set __qualname__ in all wrappers that need it.

上级 bcd856b4
......@@ -8,6 +8,7 @@ import os
import shlex
import sys
import warnings
from functools import wraps
from six import StringIO
......@@ -96,6 +97,7 @@ def change_flags(**kwargs):
Useful during tests.
"""
def change_flags_exec(f):
@wraps(f)
def inner(*args, **kwargs_):
old_val = {}
for k in kwargs:
......@@ -117,9 +119,6 @@ def change_flags(**kwargs):
assert len(l) == 1
l[0].__set__(None, old_val[k])
# Make sure that the name of the decorated function remains the same.
inner.__name__ = f.__name__
return inner
return change_flags_exec
......
......@@ -167,6 +167,8 @@ def makeTester(name, op, gpu_op, cases, checks=None, mode_gpu=mode_with_gpu,
inputs, variables))
Checker.__name__ = name
if hasattr(Checker, '__qualname__'):
Checker.__qualname__ = name
return Checker
......@@ -228,6 +230,7 @@ def gpu_alloc_expected(x, *shp):
g[:] = x
return g
GpuAllocTester = makeTester(
name="GpuAllocTester",
op=alloc,
......
......@@ -2472,6 +2472,8 @@ def _hv_switch(op, expected_function):
def expected_f(self, a, format=None, dtype=None):
return expected_function(a, format, dtype)
XStackTester.__name__ = op.__name__ + "Tester"
if hasattr(XStackTester, '__qualname__'):
XStackTester.__qualname__ = XStackTester.__name__
return XStackTester
HStackTester = _hv_switch(HStack, sp.hstack)
......@@ -2687,6 +2689,8 @@ def elemwise_checker(op, expected_f, gap=None, test_dtypes=None,
if name is None:
name = op.__name__.capitalize() + 'Tester'
Tester.__name__ = name
if hasattr(Tester, '__qualname__'):
Tester.__qualname__ = name
assert 'Roundhalftoeven' not in Tester.__name__
return Tester
......
......@@ -539,6 +539,8 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
assert None not in in_grad_vars
Checker.__name__ = name
if hasattr(Checker, '__qualname__'):
Checker.__qualname__ = name
return Checker
......
......@@ -598,6 +598,8 @@ def makeSharedTester(shared_constructor_,
assert name is not None
SharedTester.__name__ = name
if hasattr(SharedTester, '__qualname__'):
SharedTester.__qualname__ = name
return SharedTester
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论