提交 c134e609 authored 作者: ChienliMa's avatar ChienliMa

Fix a bug: While initializing a clone graph it should not be cloned again,…

Fix a bug: While initializing a clone graph it should not be cloned again, otherwise the equiv doesn't work.
上级 55277c43
...@@ -255,26 +255,17 @@ class T_function(unittest.TestCase): ...@@ -255,26 +255,17 @@ class T_function(unittest.TestCase):
fgraph_ori = ori.maker.fgraph fgraph_ori = ori.maker.fgraph
fgraph_cpy = cpy.maker.fgraph fgraph_cpy = cpy.maker.fgraph
<<<<<<< HEAD
# assert intermediate and Constants storages are shared. # assert intermediate and Constants storages are shared.
# and output stoarges are not shared # and output stoarges are not shared
=======
# assert intermediate and Constants storages are shared
>>>>>>> Add some docs to FunctionGraph and minor changes of function_module.copy()
i_o_variables = fgraph_cpy.inputs + fgraph_cpy.outputs i_o_variables = fgraph_cpy.inputs + fgraph_cpy.outputs
ori_storages = storage_map_ori.values() ori_storages = storage_map_ori.values()
for key in storage_map_cpy.keys(): for key in storage_map_cpy.keys():
storage = storage_map_cpy[key] storage = storage_map_cpy[key]
storage_is_shared = any([ storage is s for s in ori_storages]) storage_is_shared = any([ storage is s for s in ori_storages])
if key not in i_o_variables or isinstance(key, theano.tensor.Constant): if key not in i_o_variables or isinstance(key, theano.tensor.Constant):
<<<<<<< HEAD
self.assertTrue(storage_is_shared) self.assertTrue(storage_is_shared)
elif key in fgraph_cpy.outputs: elif key in fgraph_cpy.outputs:
self.assertFalse(storage_is_shared) self.assertFalse(storage_is_shared)
=======
storage = storage_map_cpy[key]
self.assertTrue( any([ storage is s for s in ori_storages]))
>>>>>>> Add some docs to FunctionGraph and minor changes of function_module.copy()
# assert storages of SharedVariable without updates are shared # assert storages of SharedVariable without updates are shared
for (input, _1, _2), here, there in zip(ori.indices, for (input, _1, _2), here, there in zip(ori.indices,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论