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

1 means preallocate as much as possible.

上级 a8719c76
...@@ -56,7 +56,7 @@ def init_dev(dev, name=None): ...@@ -56,7 +56,7 @@ def init_dev(dev, name=None):
init_dev.devmap[dev] = ctx init_dev.devmap[dev] = ctx
if config.gpuarray.preallocate > 0: if config.gpuarray.preallocate > 0:
MB = (1024 * 1024) MB = (1024 * 1024)
if config.gpuarray.preallocate < 1: if config.gpuarray.preallocate <= 1:
gmem = min(config.gpuarray.preallocate, 0.95) * ctx.total_gmem gmem = min(config.gpuarray.preallocate, 0.95) * ctx.total_gmem
else: else:
gmem = config.gpuarray.preallocate * MB gmem = config.gpuarray.preallocate * MB
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论