提交 b19eee60 authored 作者: Frederic's avatar Frederic

Don't make gpuelemwise inplace opt enabled with device=gpu and optimizer=fast_compile.

上级 4897207a
...@@ -1632,8 +1632,10 @@ else: ...@@ -1632,8 +1632,10 @@ else:
#GpuElemwise inplace #GpuElemwise inplace
gpu_inplace_elemwise_optimizer = tensor.opt.inplace_elemwise_optimizer_op( gpu_inplace_elemwise_optimizer = tensor.opt.inplace_elemwise_optimizer_op(
GpuElemwise) GpuElemwise)
# DO NOT PLACE add a 'gpu' tag here! This would enable it in fast_compile.
# It still will be run in fast_run with device=gpu with the current tag.
optdb.register('gpu_inplace_elemwise_opt', gpu_inplace_elemwise_optimizer, 75, optdb.register('gpu_inplace_elemwise_opt', gpu_inplace_elemwise_optimizer, 75,
'fast_run', 'inplace', 'gpu_inplace', 'gpu') 'fast_run', 'inplace', 'gpu_inplace')
@register_opt() @register_opt()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论