提交 2618a455 authored 作者: James Bergstra's avatar James Bergstra

removed a weird sys.exit debugging statement

上级 887947e7
...@@ -44,8 +44,8 @@ def local_gpu_elemwise_0(node): ...@@ -44,8 +44,8 @@ def local_gpu_elemwise_0(node):
if isinstance(node.op, tensor.Elemwise): if isinstance(node.op, tensor.Elemwise):
if any(hasattr(i.owner, 'op') and isinstance(i.owner.op, HostFromGpu) for i in node.inputs): if any(hasattr(i.owner, 'op') and isinstance(i.owner.op, HostFromGpu) for i in node.inputs):
if any(o.type.dtype == 'float64' for o in node.outputs): if any(o.type.dtype == 'float64' for o in node.outputs):
print 'EXITING FROM local_gpu_elemwise_0', node print 'WARNING: THERE ARE STILL float64s in your graph local_gpu_elemwise_0', node
sys.exit() else:
# move the add to a GpuAdd # move the add to a GpuAdd
new_op = GpuElemwise(node.op.scalar_op, node.op.inplace_pattern) new_op = GpuElemwise(node.op.scalar_op, node.op.inplace_pattern)
return [host_from_gpu(new_op(*(gpu_from_host(i) for i in node.inputs)))] return [host_from_gpu(new_op(*(gpu_from_host(i) for i in node.inputs)))]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论