提交 45c3b810 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Add local_useless_elemwise from tensor to clean up shape graphs from cudnn.

上级 ecfc65ec
...@@ -756,6 +756,8 @@ inplace_gpu_elemwise_opt = tensor.opt.InplaceElemwiseOptimizer( ...@@ -756,6 +756,8 @@ inplace_gpu_elemwise_opt = tensor.opt.InplaceElemwiseOptimizer(
optdb.register('gpua_inplace_opt', inplace_gpu_elemwise_opt, 75, optdb.register('gpua_inplace_opt', inplace_gpu_elemwise_opt, 75,
'inplace_elemwise_optimizer', 'fast_run', 'inplace', 'gpuarray') 'inplace_elemwise_optimizer', 'fast_run', 'inplace', 'gpuarray')
register_opt(tensor.opt.local_useless_elemwise)
@register_opt('fast_compile') @register_opt('fast_compile')
@op_lifter([tensor.DimShuffle]) @op_lifter([tensor.DimShuffle])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论