提交 18934350 authored 作者: Frederic Bastien's avatar Frederic Bastien

Keep more values_eq_approx

上级 e876a13b
......@@ -194,7 +194,11 @@ def safe_to_gpu(x, ctx_name):
def safe_to_cpu(x):
if isinstance(x.type, GpuArrayType):
return x.transfer('cpu')
ret = x.transfer('cpu')
values_eq_approx = getattr(x.tag, 'values_eq_approx', None)
if values_eq_approx:
ret.tag.values_eq_approx = values_eq_approx
return ret
else:
return x
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论