提交 36786abe authored 作者: Frederic Bastien's avatar Frederic Bastien

small fix to verbose version.

上级 586eccbb
...@@ -683,7 +683,7 @@ class NaiveAlgo(object): ...@@ -683,7 +683,7 @@ class NaiveAlgo(object):
""" %locals() """ %locals()
print >> sio, 'std::cerr << ' + " << ' ' << ".join(['" "']+list("dims[%i]"%di print >> sio, 'std::cerr << ' + " << ' ' << ".join(['" "']+list("dims[%i]"%di
for di in xrange(nd)) + ["'\\n';"]) for di in xrange(nd)) + ["'\\n';"])
if self.verbose>2: if self.verbose>1:
for ipos in xrange(len(node.inputs)): for ipos in xrange(len(node.inputs)):
print >> sio, """ print >> sio, """
std::cerr << " %(ipos)s data strides" << std::cerr << " %(ipos)s data strides" <<
...@@ -711,7 +711,7 @@ if(nd_collapse_%(ipos)s[i]==0) ...@@ -711,7 +711,7 @@ if(nd_collapse_%(ipos)s[i]==0)
nd_collapse_[i]=0; nd_collapse_[i]=0;
} }
""" %locals() """ %locals()
if self.verbose>2: if self.verbose>1:
print >>sio, """ print >>sio, """
std::cerr<< "nd_collapse_%(ipos)s "<< std::cerr<< "nd_collapse_%(ipos)s "<<
"""%locals() """%locals()
...@@ -749,7 +749,7 @@ nd_collapse_[i]=0; ...@@ -749,7 +749,7 @@ nd_collapse_[i]=0;
"""%locals() """%locals()
if self.verbose>2: if self.verbose>1:
for d in xrange(nd): for d in xrange(nd):
print >> sio, 'std::cerr << "local_dims %(d)s " << local_dims[%(d)s] << "\\n"; '%locals() print >> sio, 'std::cerr << "local_dims %(d)s " << local_dims[%(d)s] << "\\n"; '%locals()
...@@ -789,7 +789,7 @@ nd_collapse_[i]=0; ...@@ -789,7 +789,7 @@ nd_collapse_[i]=0;
} }
"""%locals() """%locals()
if self.verbose>2: if self.verbose>1:
for ipos in ["i"+ str(x) for x in xrange(len(node.inputs))]+["o"+ str(x) for x in xrange(len(node.outputs))]: for ipos in ["i"+ str(x) for x in xrange(len(node.inputs))]+["o"+ str(x) for x in xrange(len(node.outputs))]:
print >> sio, 'std::cerr << " local_%(ipos)s_str " <<'%locals()+' << " " << '.join(["local_%(ipos)s_str[%(x)s]"%locals() for x in range(nd)])+'<<"\\n";' print >> sio, 'std::cerr << " local_%(ipos)s_str " <<'%locals()+' << " " << '.join(["local_%(ipos)s_str[%(x)s]"%locals() for x in range(nd)])+'<<"\\n";'
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论