提交 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
import time
from six import iteritems
from six.moves import xrange
import sys
import theano
from theano import tensor, scalar, gof, config
......@@ -485,6 +486,15 @@ class GraphToGPU(NavigatorOptimizer):
node_created,
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])
def local_cut_gpu_transfers(node):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论