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