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

Fix mistake.

上级 5f3c1916
...@@ -459,7 +459,7 @@ class GpuAlloc(HideC, Alloc): ...@@ -459,7 +459,7 @@ class GpuAlloc(HideC, Alloc):
Parameters Parameters
---------- ----------
context : context name context_name : str
The name of the context in which to allocate memory The name of the context in which to allocate memory
memset_0 : bool memset_0 : bool
It's only an optimized version. True, it means the It's only an optimized version. True, it means the
...@@ -484,8 +484,7 @@ class GpuAlloc(HideC, Alloc): ...@@ -484,8 +484,7 @@ class GpuAlloc(HideC, Alloc):
m = "{memset_0=True}" m = "{memset_0=True}"
else: else:
m = "" m = ""
return "%s<%s>{memset_0=%s}" % (self.__class__.__name__, return "%s<%s>%s" % (self.__class__.__name__, self.context_name, m)
self.context_name, m)
def make_node(self, value, *shape): def make_node(self, value, *shape):
value = as_gpuarray_variable(value, context_name=self.context_name) value = as_gpuarray_variable(value, context_name=self.context_name)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论