提交 94732dcd authored 作者: Melanie Ducoffe's avatar Melanie Ducoffe

test optimisation GpuAllocEmpty, have to correct blancks spaces

上级 a58fb01d
...@@ -2279,6 +2279,7 @@ def gpuScanOptimization(node): ...@@ -2279,6 +2279,7 @@ def gpuScanOptimization(node):
@register_opt() @register_opt()
@local_optimizer([tensor.AllocEmpty, gpu_from_host]) @local_optimizer([tensor.AllocEmpty, gpu_from_host])
def local_gpu_allocempty(node): def local_gpu_allocempty(node):
#print 'kikou'
if (isinstance(node.op, tensor.AllocEmpty) and if (isinstance(node.op, tensor.AllocEmpty) and
node.op.dtype=="float32"): node.op.dtype=="float32"):
return [host_from_gpu(GpuAllocEmpty()(*node.inputs))] return [host_from_gpu(GpuAllocEmpty()(*node.inputs))]
......
...@@ -5544,7 +5544,6 @@ class AllocEmpty(gof.Op): ...@@ -5544,7 +5544,6 @@ class AllocEmpty(gof.Op):
} }
} }
""" % locals() """ % locals()
print str
return str return str
def infer_shape(self, node, input_shapes): def infer_shape(self, node, input_shapes):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论