提交 93add86b authored 作者: Razvan Pascanu's avatar Razvan Pascanu

PEP8 fixes

上级 f1b4a568
...@@ -1514,11 +1514,10 @@ def gpuScanOptimization(node): ...@@ -1514,11 +1514,10 @@ def gpuScanOptimization(node):
typeConstructor = lambda broadcastable, dtype: CudaNdarrayType( typeConstructor = lambda broadcastable, dtype: CudaNdarrayType(
broadcastable=broadcastable) broadcastable=broadcastable)
_outputs = scan_op.Scan( _outputs = scan_op.Scan(
scan_ins, scan_ins,
scan_outs, scan_outs,
info, info,
typeConstructor=typeConstructor).make_node( typeConstructor=typeConstructor).make_node(*nw_ins).outputs
*nw_ins).outputs
outputs = [] outputs = []
for x, y in zip(_outputs, node.outputs): for x, y in zip(_outputs, node.outputs):
if isinstance(y.type, CudaNdarrayType): if isinstance(y.type, CudaNdarrayType):
......
...@@ -307,7 +307,6 @@ scan_seqopt.register('scanOp_pushout_nonseqs_ops', ...@@ -307,7 +307,6 @@ scan_seqopt.register('scanOp_pushout_nonseqs_ops',
'scan') 'scan')
class ScanInplaceOptimizer(Optimizer): class ScanInplaceOptimizer(Optimizer):
"""Graph optimizer for Scan(makes it run inplace)""" """Graph optimizer for Scan(makes it run inplace)"""
def __init__(self, typeConstructor=None, gpu_flag=False): def __init__(self, typeConstructor=None, gpu_flag=False):
...@@ -324,7 +323,7 @@ class ScanInplaceOptimizer(Optimizer): ...@@ -324,7 +323,7 @@ class ScanInplaceOptimizer(Optimizer):
nodes = env.toposort() nodes = env.toposort()
scan_nodes = [x for x in nodes scan_nodes = [x for x in nodes
if (isinstance(x.op, scan_op.Scan) and if (isinstance(x.op, scan_op.Scan) and
x.op.info['gpu']== self.gpu_flag)] x.op.info['gpu'] == self.gpu_flag)]
for scan_idx in xrange(len(scan_nodes)): for scan_idx in xrange(len(scan_nodes)):
node = scan_nodes[scan_idx] node = scan_nodes[scan_idx]
op = node.op op = node.op
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论