提交 15a77517 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Re-add the missing single stream argument.

上级 64857efa
...@@ -87,6 +87,7 @@ def init_dev(dev, name=None, preallocate=None): ...@@ -87,6 +87,7 @@ def init_dev(dev, name=None, preallocate=None):
context = pygpu.init( context = pygpu.init(
dev, dev,
sched=config.gpuarray.sched, sched=config.gpuarray.sched,
single_stream=config.gpuarray.single_stream,
**args) **args)
context.dev = dev context.dev = dev
init_dev.devmap[dev] = context init_dev.devmap[dev] = context
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论