提交 7a87a567 authored 作者: James Bergstra's avatar James Bergstra

nicer impl of _may_share_memory in DebugMode

上级 4e61200c
......@@ -450,9 +450,13 @@ def _check_viewmap(node, storage_map):
raise BadViewMap(node, oi, outstorage, out_alias_idx=other_oi)
def _may_share_memory(a, b):
return (hasattr(a,'__array_interface__') and
hasattr(b,'__array_interface__') and
numpy.may_share_memory(a,b))
try:
return numpy.may_share_memory(a, b)
except:
return False
#return (hasattr(a,'__array_interface__') and
#hasattr(b,'__array_interface__') and
#numpy.may_share_memory(a,b))
def _is_function_output(node):
"""
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论