提交 23e9452b authored 作者: Frederic's avatar Frederic

remove the inplace flags. It is the opt that should add them, but opt in this…

remove the inplace flags. It is the opt that should add them, but opt in this file only move the computation to the gpu. They don't make them inplace.
上级 a0a5290b
...@@ -62,7 +62,7 @@ optdb.register('gpu_after_fusion', ...@@ -62,7 +62,7 @@ optdb.register('gpu_after_fusion',
def register_opt(*tags, **kwargs): def register_opt(*tags, **kwargs):
def f(local_opt): def f(local_opt):
name = (kwargs and kwargs.pop('name')) or local_opt.__name__ name = (kwargs and kwargs.pop('name')) or local_opt.__name__
gpu_optimizer.register(name, local_opt, 'fast_run', 'inplace', *tags) gpu_optimizer.register(name, local_opt, 'fast_run', *tags)
return local_opt return local_opt
return f return f
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论