提交 4df20428 authored 作者: Nicolas Ballas's avatar Nicolas Ballas

Update print output of storage map

上级 f3e696a5
...@@ -181,16 +181,20 @@ def raise_with_op(node, thunk=None, exc_info=None, storage_map=None): ...@@ -181,16 +181,20 @@ def raise_with_op(node, thunk=None, exc_info=None, storage_map=None):
shapeinfo = None shapeinfo = None
if hasattr(storage_map[k][0], 'shape'): if hasattr(storage_map[k][0], 'shape'):
shapeinfo = storage_map[k][0].shape shapeinfo = storage_map[k][0].shape
detailed_err_msg += "shape: %s, " % str(shapeinfo) if len(shapeinfo) != 0:
detailed_err_msg += "Shape: %s, " % str(shapeinfo)
else:
detailed_err_msg += "Shape: (1,), "
if hasattr(storage_map[k][0], 'dtype'): if hasattr(storage_map[k][0], 'dtype'):
dtype = storage_map[k][0].dtype dtype = storage_map[k][0].dtype
detailed_err_msg += "ElemSize: %s Byte(s)" % numpy.dtype(dtype).itemsize
if shapeinfo is None: if shapeinfo is None:
detailed_err_msg += "size: %s Byte(s)\n" % numpy.dtype(dtype).itemsize detailed_err_msg += "\n"
else: else:
detailed_err_msg += "size: %s Byte(s)\n" % (numpy.dtype(dtype).itemsize * numpy.prod(shapeinfo)) detailed_err_msg += ", TotalSize: %s Byte(s)\n" % (numpy.dtype(dtype).itemsize * numpy.prod(shapeinfo))
else: else:
bytes = getsizeof(storage_map[k][0]) bytes = getsizeof(storage_map[k][0])
detailed_err_msg += "elementsize: %s Byte(s)\n" % str(bytes) detailed_err_msg += "ElemSize: %s Byte(s)\n" % str(bytes)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论