提交 c9187dec authored 作者: Ziye Fan's avatar Ziye Fan

let gpu_merge and gpu_constant_folding be final_opt

上级 ed31d840
...@@ -94,13 +94,13 @@ optdb.register('gpu_after_fusion', ...@@ -94,13 +94,13 @@ optdb.register('gpu_after_fusion',
# Register merge_optimizer as a global opt # Register merge_optimizer as a global opt
gpu_optimizer.register('gpu_merge', theano.gof.opt.merge_optimizer, gpu_optimizer.register('gpu_merge', theano.gof.opt.merge_optimizer,
'fast_run', 'fast_compile') 'fast_run', 'fast_compile', 'final_opt')
# register local_track_shape_i at this level too # register local_track_shape_i at this level too
# to make multi-level lift of shape work. # to make multi-level lift of shape work.
register_opt()(theano.tensor.opt.local_track_shape_i) register_opt()(theano.tensor.opt.local_track_shape_i)
register_opt(name='gpu_constant_folding')( register_opt('final_opt', name='gpu_constant_folding')(
tensor.opt.constant_folding) tensor.opt.constant_folding)
register_opt()(theano.tensor.opt.local_subtensor_make_vector) register_opt()(theano.tensor.opt.local_subtensor_make_vector)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论