提交 301c4a23 authored 作者: Frederic's avatar Frederic

Fix daily build test due to rounding problems

上级 41075539
...@@ -235,7 +235,8 @@ def upcast_int8_nfunc(fn): ...@@ -235,7 +235,8 @@ 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):
""" """
:param check_name: :param check_name:
Use only for tester that aren't in Theano. Use only for tester that aren't in Theano.
...@@ -258,6 +259,7 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None, ...@@ -258,6 +259,7 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
_mode, _grad_rtol, _eps, skip_ = mode, grad_rtol, eps, skip _mode, _grad_rtol, _eps, skip_ = mode, grad_rtol, eps, skip
_test_memmap = test_memmap _test_memmap = test_memmap
_check_name = check_name _check_name = check_name
_grad_eps = grad_eps
class Checker(unittest.TestCase): class Checker(unittest.TestCase):
...@@ -463,8 +465,9 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None, ...@@ -463,8 +465,9 @@ def makeTester(name, op, expected, checks=None, good=None, bad_build=None,
inputs = [copy(input) for input in inputs] inputs = [copy(input) for input in inputs]
try: try:
utt.verify_grad(self.op, inputs, utt.verify_grad(self.op, inputs,
mode=self.mode, mode=self.mode,
rel_tol=_grad_rtol) rel_tol=_grad_rtol,
eps=_grad_eps)
except Exception, exc: except Exception, exc:
err_msg = ("Test %s::%s: Error occurred while" err_msg = ("Test %s::%s: Error occurred while"
" computing the gradient on the following" " computing the gradient on the following"
...@@ -976,6 +979,7 @@ ModTester = makeBroadcastTester( ...@@ -976,6 +979,7 @@ ModTester = makeBroadcastTester(
good=copymod(_good_broadcast_div_mod_normal_float, good=copymod(_good_broadcast_div_mod_normal_float,
['complex1', 'complex2']), ['complex1', 'complex2']),
grad=_grad_broadcast_div_mod_normal, grad=_grad_broadcast_div_mod_normal,
grad_eps=1e-5,
) )
...@@ -986,6 +990,7 @@ ModInplaceTester = makeBroadcastTester( ...@@ -986,6 +990,7 @@ ModInplaceTester = makeBroadcastTester(
good=copymod(_good_broadcast_div_mod_normal_float_inplace, good=copymod(_good_broadcast_div_mod_normal_float_inplace,
["complex1", "complex2"]), ["complex1", "complex2"]),
grad=_grad_broadcast_div_mod_normal, grad=_grad_broadcast_div_mod_normal,
grad_eps=1e-5,
inplace=True) inplace=True)
_good_broadcast_pow_normal_float = dict(same_shapes=(rand_ranged(1, 5, (2, 3)), rand_ranged(-3, 3, (2, 3))), _good_broadcast_pow_normal_float = dict(same_shapes=(rand_ranged(1, 5, (2, 3)), rand_ranged(-3, 3, (2, 3))),
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论