提交 6fc0bd7d authored 作者: Alassane Ndiaye's avatar Alassane Ndiaye

Added check in order to see if 'may_share_memory' exists.

上级 b0f7fecb
...@@ -768,8 +768,8 @@ def _check_viewmap(node, storage_map): ...@@ -768,8 +768,8 @@ def _check_viewmap(node, storage_map):
# case... # case...
for ii, inode in enumerate(node.inputs): for ii, inode in enumerate(node.inputs):
if hasattr(inode.type, 'may_share_memory') and\
if inode.type.may_share_memory(outstorage, storage_map[inode][0]): inode.type.may_share_memory(outstorage, storage_map[inode][0]):
nodeid = id(inode) nodeid = id(inode)
bad_alias[nodeid] = ii bad_alias[nodeid] = ii
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论