提交 72997cde authored 作者: Frederic's avatar Frederic

small crash fix in new pickling of shared var on gpu.

上级 507c5a3e
...@@ -288,7 +288,7 @@ def handle_shared_float32(tf): ...@@ -288,7 +288,7 @@ def handle_shared_float32(tf):
raise NotImplementedError('removing our handler') raise NotImplementedError('removing our handler')
def reduce_tensor_variable(var): def reduce_tensor_variable(var):
if isinstance(var.owner.op, HostFromGpu) and len(var.owner.inputs) == 1 \ if var.owner and isinstance(var.owner.op, HostFromGpu) and len(var.owner.inputs) == 1 \
and isinstance(var.owner.inputs[0], CudaNdarraySharedVariable): and isinstance(var.owner.inputs[0], CudaNdarraySharedVariable):
return load_shared_variable, (var.owner.inputs[0].get_value(),) return load_shared_variable, (var.owner.inputs[0].get_value(),)
else: else:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论