提交 4a51ae52 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

This needed to ba able to use the old and new backends at the same time.

上级 6a3cb283
...@@ -70,12 +70,12 @@ def local_cut_gpu_host_gpu(node): ...@@ -70,12 +70,12 @@ def local_cut_gpu_host_gpu(node):
if tensor.opt.opt.check_chain(node, host_from_gpu, gpu_from_host): if tensor.opt.opt.check_chain(node, host_from_gpu, gpu_from_host):
return [node.inputs[0].owner.inputs[0]] return [node.inputs[0].owner.inputs[0]]
return False return False
gpu_cut_copies.register('cut_gpu_host_transfers', local_cut_gpu_host_gpu, gpu_cut_copies.register('cut_gpua_host_transfers', local_cut_gpu_host_gpu,
'fast_run', 'inplace', 'gpuarray') 'fast_run', 'inplace', 'gpuarray')
gpu_cut_copies.register('cut_gpu_constant_transfers', gpu_cut_copies.register('cut_gpua_constant_transfers',
tensor.opt.constant_folding, tensor.opt.constant_folding,
'fast_run', 'gpuarray') 'fast_run', 'gpuarray')
optdb['canonicalize'].register('local_cut_gpu_host_gpu', optdb['canonicalize'].register('local_cut_gpua_host_gpua',
local_cut_gpu_host_gpu, 'fast_run', 'gpuarray') local_cut_gpu_host_gpu, 'fast_run', 'gpuarray')
@register_opt() @register_opt()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论