提交 c8fdd9f1 authored 作者: Frederic's avatar Frederic

Better erro message and clean the returned dict

上级 512925b8
...@@ -1742,7 +1742,7 @@ def get_device_type_sizes(): ...@@ -1742,7 +1742,7 @@ def get_device_type_sizes():
cuda_ndarray = theano.sandbox.cuda.cuda_ndarray.cuda_ndarray cuda_ndarray = theano.sandbox.cuda.cuda_ndarray.cuda_ndarray
t = cuda_ndarray.ptr_int_size() t = cuda_ndarray.ptr_int_size()
gpu_ptr_size, cpu_ptr_size, int_size, gpu_int_size = t gpu_ptr_size, cpu_ptr_size, int_size, gpu_int_size = t
assert int_size == gpu_int_size assert int_size == gpu_int_size, (int_size, gpu_int_size)
del gpu_int_size del gpu_int_size
del t del t
except Exception, e: except Exception, e:
...@@ -1751,7 +1751,9 @@ def get_device_type_sizes(): ...@@ -1751,7 +1751,9 @@ def get_device_type_sizes():
"This could cause less GpuElemwise fused together.\n" "This could cause less GpuElemwise fused together.\n"
"%s") % e) "%s") % e)
rval = get_device_type_sizes.rval = locals() rval = get_device_type_sizes.rval = dict(gpu_ptr_size=gpu_ptr_size,
cpu_ptr_size=cpu_ptr_size,
int_size=int_size)
return rval return rval
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论