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

test with smaller size with the python linker.

上级 757a24d5
...@@ -359,7 +359,8 @@ def test_uniform(): ...@@ -359,7 +359,8 @@ def test_uniform():
#TODO: test ndim!=size.ndim #TODO: test ndim!=size.ndim
#TODO: test bad seed #TODO: test bad seed
#TODO: test size=Var, with shape that change from call to call #TODO: test size=Var, with shape that change from call to call
if mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE']: if (mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE'] or
mode == 'Mode' and config.linker in ['py']):
sample_size = (10, 100) sample_size = (10, 100)
steps = 50 steps = 50
else: else:
...@@ -440,7 +441,8 @@ def test_binomial(): ...@@ -440,7 +441,8 @@ def test_binomial():
#we test size in a tuple of int and a tensor.shape. #we test size in a tuple of int and a tensor.shape.
#we test the param p with int. #we test the param p with int.
if mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE']: if (mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE'] or
mode == 'Mode' and config.linker in ['py']):
sample_size = (10, 50) sample_size = (10, 50)
steps = 50 steps = 50
rtol = 0.02 rtol = 0.02
...@@ -510,7 +512,8 @@ def test_normal0(): ...@@ -510,7 +512,8 @@ def test_normal0():
steps = 50 steps = 50
std = 2. std = 2.
if mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE']: if (mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE'] or
mode == 'Mode' and config.linker in ['py']):
sample_size = (25, 30) sample_size = (25, 30)
default_rtol = .02 default_rtol = .02
else: else:
...@@ -617,7 +620,8 @@ def test_multinomial(): ...@@ -617,7 +620,8 @@ def test_multinomial():
if mode == 'FAST_COMPILE': if mode == 'FAST_COMPILE':
mode_ = 'FAST_RUN' mode_ = 'FAST_RUN'
if mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE']: if (mode in ['DEBUG_MODE', 'DebugMode', 'FAST_COMPILE'] or
mode == 'Mode' and config.linker in ['py']):
sample_size = (49, 5) sample_size = (49, 5)
else: else:
sample_size = (450, 6) sample_size = (450, 6)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论