提交 2aec1bab authored 作者: Frederic Bastien's avatar Frederic Bastien

rename opt and add comment

上级 b4aefd7d
...@@ -23,7 +23,7 @@ theano.compile.optdb.register('local_inplace_remove0', ...@@ -23,7 +23,7 @@ theano.compile.optdb.register('local_inplace_remove0',
@gof.local_optimizer([csm_properties]) @gof.local_optimizer([csm_properties])
def skip_pack_csc01(node): def local_csm_properties_csm(node):
"""if we find csm_properties(CSM(*args)), then we can replace that with the """if we find csm_properties(CSM(*args)), then we can replace that with the
*args directly""" *args directly"""
if node.op == csm_properties: if node.op == csm_properties:
...@@ -36,4 +36,4 @@ def skip_pack_csc01(node): ...@@ -36,4 +36,4 @@ def skip_pack_csc01(node):
return ret_var return ret_var
return False return False
register_specialize(skip_pack_csc01) register_specialize(local_csm_properties_csm)
...@@ -1188,7 +1188,8 @@ def test_remove0(): ...@@ -1188,7 +1188,8 @@ def test_remove0():
f = theano.function([theano.In(x, borrow=True, mutable=True)], f = theano.function([theano.In(x, borrow=True, mutable=True)],
Remove0()(x)) Remove0()(x))
# assert optimization is applied in modes with optimization # assert optimization local_inplace_remove0 is applied in
# modes with optimization
if theano.config.mode not in ['FAST_COMPILE']: if theano.config.mode not in ['FAST_COMPILE']:
# list of apply nodes in the optimized graph. # list of apply nodes in the optimized graph.
nodes = f.maker.env.toposort() nodes = f.maker.env.toposort()
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论