提交 885142fc authored 作者: James Bergstra's avatar James Bergstra

renamed test_speed -> compare_speed

上级 96a4acd5
...@@ -29,6 +29,7 @@ def test_uniform_basic(): ...@@ -29,6 +29,7 @@ def test_uniform_basic():
assert v.min() < v.max() assert v.min() < v.max()
assert .25 <= v.mean() <= .75 assert .25 <= v.mean() <= .75
def test_normal_basic(): def test_normal_basic():
rng = CURAND_RandomStreams(234) rng = CURAND_RandomStreams(234)
...@@ -53,7 +54,13 @@ def test_normal_basic(): ...@@ -53,7 +54,13 @@ def test_normal_basic():
assert v.min() < v.max() assert v.min() < v.max()
assert -.5 <= v.mean() <= .5 assert -.5 <= v.mean() <= .5
def test_speed():
def compare_speed():
# To run this speed comparison
# cd <directory of this file>
# THEANO_FLAGS=device=gpu \
# python -c 'import test_rng_curand; test_rng_curand.compare_speed()'
mrg = MRG_RandomStreams() mrg = MRG_RandomStreams()
crn = CURAND_RandomStreams(234) crn = CURAND_RandomStreams(234)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论