提交 d4a825d6 authored 作者: Frederic's avatar Frederic

Fix and enable Chi2SF[Inplace]Tester tests.

they work on the GPU.
上级 8fd8657e
...@@ -1421,7 +1421,7 @@ if imported_scipy_special: ...@@ -1421,7 +1421,7 @@ if imported_scipy_special:
expected_gamma = scipy.special.gamma expected_gamma = scipy.special.gamma
expected_gammaln = scipy.special.gammaln expected_gammaln = scipy.special.gammaln
expected_psi = scipy.special.psi expected_psi = scipy.special.psi
expected_chi2sf = scipy.stats.chi2.sf expected_chi2sf = lambda x, df: scipy.stats.chi2.sf(x, df).astype(x.dtype)
skip_scipy = False skip_scipy = False
else: else:
expected_erf = [] expected_erf = []
...@@ -1552,13 +1552,13 @@ PsiInplaceTester = makeBroadcastTester( ...@@ -1552,13 +1552,13 @@ PsiInplaceTester = makeBroadcastTester(
skip=skip_scipy) skip=skip_scipy)
'''
#chi2sf takes two inputs, a value (x) and a degrees of freedom (k). #chi2sf takes two inputs, a value (x) and a degrees of freedom (k).
# not sure how to deal with that here... # not sure how to deal with that here...
_good_broadcast_unary_chi2sf = dict( _good_broadcast_unary_chi2sf = dict(
normal=(rand_ranged(1, 10, (2, 3)),), normal=(rand_ranged(1, 10, (2, 3)), numpy.asarray(1, dtype=config.floatX)),
empty=(numpy.asarray([]),),) empty=(numpy.asarray([], dtype=config.floatX),
numpy.asarray(1, dtype=config.floatX)))
Chi2SFTester = makeBroadcastTester( Chi2SFTester = makeBroadcastTester(
op=tensor.chi2sf, op=tensor.chi2sf,
...@@ -1566,7 +1566,8 @@ Chi2SFTester = makeBroadcastTester( ...@@ -1566,7 +1566,8 @@ Chi2SFTester = makeBroadcastTester(
good=_good_broadcast_unary_chi2sf, good=_good_broadcast_unary_chi2sf,
eps=2e-10, eps=2e-10,
mode=mode_no_scipy, mode=mode_no_scipy,
skip=skip_scipy) skip=skip_scipy,
name='Chi2SF')
Chi2SFInplaceTester = makeBroadcastTester( Chi2SFInplaceTester = makeBroadcastTester(
op=inplace.chi2sf_inplace, op=inplace.chi2sf_inplace,
expected=expected_chi2sf, expected=expected_chi2sf,
...@@ -1574,8 +1575,8 @@ Chi2SFInplaceTester = makeBroadcastTester( ...@@ -1574,8 +1575,8 @@ Chi2SFInplaceTester = makeBroadcastTester(
eps=2e-10, eps=2e-10,
mode=mode_no_scipy, mode=mode_no_scipy,
inplace=True, inplace=True,
skip=skip_scipy) skip=skip_scipy,
''' name='Chi2SF')
ZerosLikeTester = makeBroadcastTester( ZerosLikeTester = makeBroadcastTester(
op=tensor.zeros_like, op=tensor.zeros_like,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论