提交 18fb4acc authored 作者: Frederic's avatar Frederic

Do not try to use Elemwise inplace opt on GPUAElemwise

上级 c5b8fcf6
......@@ -233,7 +233,8 @@ def inplace_elemwise_optimizer_op(OP):
for node in list(graph.io_toposort(fgraph.inputs, fgraph.outputs)):
op = node.op
if not isinstance(op, OP):
# gpuarray GpuElemwise inherit from Elemwise
if not type(op) == OP:
continue
baseline = op.inplace_pattern
protected_inputs = [
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论