提交 a9461657 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Cosmetic changes

上级 5281bc31
......@@ -562,13 +562,13 @@ def local_track_shape_i(node):
shape_feature = node.env.shape_feature
except:
return
if node in node.env.shape_feature.scheduled:
if node in shape_feature.scheduled:
assert isinstance(node.op, Shape_i)
replacement = node.env.shape_feature.scheduled[node]
#if len(node.env.shape_feature.shape_of[replacement])!=1:
replacement = shape_feature.scheduled[node]
#if len(shape_feature.shape_of[replacement])!=1:
#import pdb;pdb.set_trace()
#print >> sys.stderr, "REPLACING SOMETHING", replacement, node.env.shape_feature.shape_of[replacement]
return [node.env.shape_feature.shape_of[replacement][0]]
print >> sys.stderr, "REPLACING SOMETHING", replacement, shape_feature.shape_of[replacement]
return [shape_feature.shape_of[replacement][node.op.i]]
@register_specialize
@register_canonicalize
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论