提交 b70ba545 authored 作者: Pierre Luc Carrier's avatar Pierre Luc Carrier

Add description for test

上级 08b014be
...@@ -2698,6 +2698,11 @@ class T_Scan(unittest.TestCase): ...@@ -2698,6 +2698,11 @@ class T_Scan(unittest.TestCase):
utt.assert_allclose(f_vals, f2_vals) utt.assert_allclose(f_vals, f2_vals)
def test_gpu_memory_usage(self): def test_gpu_memory_usage(self):
# This test validates that the memory usage of the defined theano
# function is reasonnable when executed on the GPU. It checks for
# a bug in which one of scan's optimization was not applied which
# made the scan node compute large and unnecessary outputs which
# brought memory usage on the GPU to ~12G.
# The test must be performed on the GPU # The test must be performed on the GPU
from theano.sandbox import cuda from theano.sandbox import cuda
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论