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