提交 b8ce8ec1 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

merged

...@@ -53,6 +53,10 @@ def rebuild_collect_shared( outputs ...@@ -53,6 +53,10 @@ def rebuild_collect_shared( outputs
"updates" nor in "no_default_updates". "updates" nor in "no_default_updates".
""" """
if isinstance(outputs,tuple):
outputs = list(outputs)
## This function implements similar functionality as graph.clone ## This function implements similar functionality as graph.clone
## and it should be merged with that ## and it should be merged with that
clone_d = {} clone_d = {}
......
...@@ -49,7 +49,7 @@ void * device_malloc(size_t size) ...@@ -49,7 +49,7 @@ void * device_malloc(size_t size)
#else #else
fprintf(stderr, "Error allocating %li bytes of device memory (%s).\n", (long)size, cudaGetErrorString(err)); fprintf(stderr, "Error allocating %li bytes of device memory (%s).\n", (long)size, cudaGetErrorString(err));
#endif #endif
PyErr_Format(PyExc_MemoryError, "error allocating %li bytes of device memory (%s)", (long)size, cudaGetErrorString(err)); PyErr_Format(PyExc_MemoryError, "Error allocating %%li bytes of device memory (%%s).", (long)size, cudaGetErrorString(err));
return NULL; return NULL;
} }
_outstanding_mallocs[0] += (rval != NULL); _outstanding_mallocs[0] += (rval != NULL);
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论