提交 5b906ea5 authored 作者: Pascal Lamblin's avatar Pascal Lamblin

Fix in erfinv GPU optimization

上级 a8865593
......@@ -241,7 +241,7 @@ def local_gpu_elemwise_1(node):
# Don't set any inplace pattern.
# gpu_inplace_elemwise_optimizer will do it later
if isinstance(node.op.scalar_op, Erfinv):
if isinstance(elemwise_node.op.scalar_op, Erfinv):
new_op = GpuElemwise(erfinv_gpu)
else:
try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论