提交 c691a273 authored 作者: Frederic Bastien's avatar Frederic Bastien

Add GraphToGPU.print_summary. We can't use the NavigationOptimizer version.

上级 ee29c769
...@@ -6,6 +6,7 @@ import pdb ...@@ -6,6 +6,7 @@ import pdb
import time import time
from six import iteritems from six import iteritems
from six.moves import xrange from six.moves import xrange
import sys
import theano import theano
from theano import tensor, scalar, gof, config from theano import tensor, scalar, gof, config
...@@ -485,6 +486,15 @@ class GraphToGPU(NavigatorOptimizer): ...@@ -485,6 +486,15 @@ class GraphToGPU(NavigatorOptimizer):
node_created, node_created,
process_count) process_count)
def print_summary(self, stream=sys.stdout, level=0, depth=-1):
print("%s%s (%i)" % (
(' ' * level), self.__class__.__name__, id(self)), file=stream)
if depth != 0:
for opt in (self.local_optimizers_all +
reduce(list.__add__,
self.local_optimizers_map.values())):
opt.print_summary(stream, level=(level + 2), depth=(depth - 1))
@local_optimizer([GpuFromHost, GpuToGpu, HostFromGpu]) @local_optimizer([GpuFromHost, GpuToGpu, HostFromGpu])
def local_cut_gpu_transfers(node): def local_cut_gpu_transfers(node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论