提交 f6593660 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Change phrasing to be clearer in the single_stream flag doc and add it to the docs.

上级 41f87ccc
...@@ -485,6 +485,21 @@ import theano and print the config variable, as in: ...@@ -485,6 +485,21 @@ import theano and print the config variable, as in:
automatically to get more memory. But this can cause automatically to get more memory. But this can cause
fragmentation, see note above. fragmentation, see note above.
.. attribute:: config.gpuarray.single_stream
Boolean value
Default: ``True``
Control the stream mode of contexts.
If your computations are mostly lots of small elements, using
single-stream will avoid the synchronization overhead and usually
be faster. For larger elements it does not make a difference yet.
In the future when true multi-stream is enabled in libgpuarray,
this may change. If you want to make sure to have optimal
performance, check both options.
.. attribute:: linker .. attribute:: linker
......
...@@ -243,9 +243,14 @@ AddConfigVar('gpuarray.preallocate', ...@@ -243,9 +243,14 @@ AddConfigVar('gpuarray.preallocate',
in_c_key=False) in_c_key=False)
AddConfigVar('gpuarray.single_stream', AddConfigVar('gpuarray.single_stream',
"""Switch between single stream mode or multi-stream """
mode. If your computation can't benefit from multiple If your computations are mostly lots of small elements,
streams, single-stream is usually faster (by about 10%) using single-stream will avoid the synchronization
overhead and usually be faster. For larger elements it
does not make a difference yet. In the future when true
multi-stream is enabled in libgpuarray, this may change.
If you want to make sure to have optimal performance,
check both options.
""", """,
BoolParam(True), BoolParam(True),
in_c_key=False) in_c_key=False)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论