提交 6a78a581 authored 作者: Shubh Vachher's avatar Shubh Vachher

skip_verify_grad added to makeTester for all InplaceTester

上级 fcbb3e9b
...@@ -249,7 +249,7 @@ def upcast_int8_nfunc(fn): ...@@ -249,7 +249,7 @@ def upcast_int8_nfunc(fn):
def makeTester(name, op, expected, checks=None, good=None, bad_build=None, def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
bad_runtime=None, grad=None, mode=None, grad_rtol=None, bad_runtime=None, grad=None, mode=None, grad_rtol=None,
eps=1e-10, skip=False, test_memmap=True, check_name=True, eps=1e-10, skip=False, test_memmap=True, check_name=True,
grad_eps=None): grad_eps=None, inplace=False):
# :param check_name: # :param check_name:
# Use only for tester that aren't in Theano. # Use only for tester that aren't in Theano.
if checks is None: if checks is None:
...@@ -260,10 +260,10 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None, ...@@ -260,10 +260,10 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
bad_build = {} bad_build = {}
if bad_runtime is None: if bad_runtime is None:
bad_runtime = {} bad_runtime = {}
if grad is None:
grad = {}
if grad is True: if grad is True:
grad = good grad = good
if grad is None or inplace is True:
grad = {}
_op, _expected, _checks, _good = op, expected, checks, good _op, _expected, _checks, _good = op, expected, checks, good
_bad_build, _bad_runtime, _grad = bad_build, bad_runtime, grad _bad_build, _bad_runtime, _grad = bad_build, bad_runtime, grad
...@@ -653,7 +653,6 @@ def makeBroadcastTester(op, expected, checks=None, name=None, **kwargs): ...@@ -653,7 +653,6 @@ def makeBroadcastTester(op, expected, checks=None, name=None, **kwargs):
return numpy.all(inputs[0] == outputs[0]) return numpy.all(inputs[0] == outputs[0])
checks = dict(checks, inplace_check=inplace_check) checks = dict(checks, inplace_check=inplace_check)
del kwargs['inplace']
return makeTester(name, op, expected, checks, **kwargs) return makeTester(name, op, expected, checks, **kwargs)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论