提交 6101abb3 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Use the gpuarray tag for gpuarray optimisations.

上级 c3557a5d
...@@ -146,13 +146,13 @@ gpu_local_elemwise_fusion = tensor.opt.local_elemwise_fusion_op( ...@@ -146,13 +146,13 @@ gpu_local_elemwise_fusion = tensor.opt.local_elemwise_fusion_op(
GpuElemwise, GpuElemwise,
max_inputs_to_GpuElemwise) max_inputs_to_GpuElemwise)
optdb.register('gpua_elemwise_fusion', optdb.register('gpua_elemwise_fusion',
tensor.opt.FusionOptimizer(gpu_local_elemwise_fusion), tensor.opt.FusionOptimizer(gpu_local_elemwise_fusion), 71.00,
71.00, 'fast_run', 'fusion', 'local_elemwise_fusion', 'gpu') 'fast_run', 'fusion', 'local_elemwise_fusion', 'gpuarray')
inplace_gpu_elemwise_opt = tensor.opt.inplace_elemwise_optimizer_op( inplace_gpu_elemwise_opt = tensor.opt.inplace_elemwise_optimizer_op(
GpuElemwise) GpuElemwise)
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') 'inplace_elemwise_optimizer', 'fast_run', 'inplace', 'gpuarray')
@register_opt() @register_opt()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论