提交 155ea053 authored 作者: Melanie Ducoffe's avatar Melanie Ducoffe

bla bla

上级 ea3e7426
......@@ -2278,7 +2278,7 @@ def gpuScanOptimization(node):
@local_optimizer([tensor.AllocEmpty, gpu_from_host])
def local_gpu_allocempty(node):
if (isinstance(node.op, tensor.AllocEmpty) and
node.op.dtype=="NPY_FLOAT_32":
node.op.dtype=="NPY_FLOAT_32"):
if any([(i.owner and isinstance(i.owner.op, HostFromGpu))
for i in node.inputs]):
return [host_from_gpu(GpuAllocEmpty("float32")(gpu_from_host(*node.inputs)))]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论