提交 33e39132 authored 作者: Frederic's avatar Frederic

round and cast to int the memory profile info.

上级 920acaa8
...@@ -605,7 +605,8 @@ class ProfileStats(object): ...@@ -605,7 +605,8 @@ class ProfileStats(object):
for fgraph, nodes_mem in fct_memory.iteritems(): for fgraph, nodes_mem in fct_memory.iteritems():
size_sum = sum([sum(val) size_sum = sum([sum(val)
for key, val in nodes_mem.iteritems()]) for key, val in nodes_mem.iteritems()])
print " Max without gc, inplace and view (KB)", size_sum / 1024 print " Max without gc, inplace and view (KB)", int(
round(size_sum / 1024))
node_memory_size = 0 node_memory_size = 0
node_memory_saved_by_view = 0 node_memory_saved_by_view = 0
...@@ -648,14 +649,16 @@ class ProfileStats(object): ...@@ -648,14 +649,16 @@ class ProfileStats(object):
pass pass
pass pass
print " Max FAST_RUN_NO_GC (KB)", node_memory_size / 1024 print " Max allow_gc=False (KB)", int(round(
print " Max c|py (KB)", running_max_memory_size / 1024 node_memory_size / 1024.))
print " Memory saved by view (KB)", ( print " Max linker=c|py (KB)", int(round(
node_memory_saved_by_view / 1024) running_max_memory_size / 1024.))
print " Memory saved by inplace (KB)", ( print " Memory saved by view (KB)", int(round(
node_memory_saved_by_inplace / 1024) node_memory_saved_by_view / 1024.))
print " Memory saved by GC (KB)", ( print " Memory saved by inplace (KB)", int(round(
node_memory_size - running_max_memory_size) / 1024 node_memory_saved_by_inplace / 1024.))
print " Memory saved by GC (KB)", int(round((
node_memory_size - running_max_memory_size) / 1024.))
print print
print " <Sum apply outputs (bytes)> <Apply outputs shape> <created/inplace/view> <Apply node>" print " <Sum apply outputs (bytes)> <Apply outputs shape> <created/inplace/view> <Apply node>"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论