提交 e6d70b6a authored 作者: Frederic Bastien's avatar Frederic Bastien

80 caracteres per line.

上级 46cd9e62
...@@ -93,8 +93,9 @@ def local_cut_gpu_host_gpu(node): ...@@ -93,8 +93,9 @@ def local_cut_gpu_host_gpu(node):
return False return False
gpu_cut_copies.register('cut_gpu_host_transfers', local_cut_gpu_host_gpu, gpu_cut_copies.register('cut_gpu_host_transfers', local_cut_gpu_host_gpu,
'fast_run', 'inplace', 'gpu') 'fast_run', 'inplace', 'gpu')
gpu_cut_copies.register('cut_gpu_constant_transfers', tensor.opt.constant_folding, gpu_cut_copies.register('cut_gpu_constant_transfers',
'fast_run', 'gpu') tensor.opt.constant_folding,
'fast_run', 'gpu')
#register it into canonicalize to allow other optimization to work without #register it into canonicalize to allow other optimization to work without
#botering with this useless pattern. #botering with this useless pattern.
compile.optdb['canonicalize'].register('local_cut_gpu_host_gpu', local_cut_gpu_host_gpu, 'fast_run') compile.optdb['canonicalize'].register('local_cut_gpu_host_gpu', local_cut_gpu_host_gpu, 'fast_run')
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论