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

Fix in erfinv GPU optimization

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