Flake8 fixes

上级 9b44a4d6
...@@ -1117,6 +1117,7 @@ def local_advincsub1_gpua_inplace(node): ...@@ -1117,6 +1117,7 @@ def local_advincsub1_gpua_inplace(node):
if not node.op.inplace: if not node.op.inplace:
return [node.op.clone_inplace()(*node.inputs)] return [node.op.clone_inplace()(*node.inputs)]
# AllocDiag # AllocDiag
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.AllocDiag]) @op_lifter([tensor.AllocDiag])
...@@ -1127,6 +1128,7 @@ def local_gpu_alloc_diag(op, context_name, inputs, outputs): ...@@ -1127,6 +1128,7 @@ def local_gpu_alloc_diag(op, context_name, inputs, outputs):
return False return False
return GpuAllocDiag(offset=op.offset) return GpuAllocDiag(offset=op.offset)
# ExtractDiag # ExtractDiag
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.ExtractDiag]) @op_lifter([tensor.ExtractDiag])
...@@ -1134,6 +1136,7 @@ def local_gpu_alloc_diag(op, context_name, inputs, outputs): ...@@ -1134,6 +1136,7 @@ def local_gpu_alloc_diag(op, context_name, inputs, outputs):
def local_gpu_extract_diag(op, context_name, inputs, outputs): def local_gpu_extract_diag(op, context_name, inputs, outputs):
return GpuExtractDiag(offset=op.offset, axis1=op.axis1, axis2=op.axis2, view=op.view) return GpuExtractDiag(offset=op.offset, axis1=op.axis1, axis2=op.axis2, view=op.view)
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.CAReduce, tensor.Sum, tensor.elemwise.Prod]) @op_lifter([tensor.CAReduce, tensor.Sum, tensor.elemwise.Prod])
@register_opt2([tensor.CAReduce, tensor.Sum, tensor.elemwise.Prod], 'fast_compile') @register_opt2([tensor.CAReduce, tensor.Sum, tensor.elemwise.Prod], 'fast_compile')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论