提交 37115ad1 authored 作者: notoraptor's avatar notoraptor

Flake8 error (again ! ...)

上级 218d006c
...@@ -1775,13 +1775,13 @@ def _scan_type_infer(node): ...@@ -1775,13 +1775,13 @@ def _scan_type_infer(node):
context_name=context_name) context_name=context_name)
return typebuild return typebuild
## Add optimization : maxandargmax (CPU -> GPU)
# Add optimization : maxandargmax (CPU -> GPU)
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.MaxAndArgmax]) @op_lifter([tensor.MaxAndArgmax])
@register_opt2([tensor.MaxAndArgmax], 'fast_compile') @register_opt2([tensor.MaxAndArgmax], 'fast_compile')
def local_gpu_maxandargmax(op, context_name, inputs, outputs): def local_gpu_maxandargmax(op, context_name, inputs, outputs):
return gpu_maxandargmax return gpu_maxandargmax
##
# Do not register in fast_run or fast_compile. # Do not register in fast_run or fast_compile.
# It will be added to fast_run if the GPU is enabled. # It will be added to fast_run if the GPU is enabled.
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论