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

Flake8 error (again ! ...)

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