From 257464ef7b3ac24ed2fb1d0760fd93b44c8bfd17 Mon Sep 17 00:00:00 2001
From: maoweiyong <maoweiyong@huawei.com>
Date: Thu, 16 Apr 2020 16:16:17 +0800
Subject: [PATCH] fix gpu op register

---
 mindspore/ops/_op_impl/akg/gpu/squeeze.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/mindspore/ops/_op_impl/akg/gpu/squeeze.py b/mindspore/ops/_op_impl/akg/gpu/squeeze.py
index 378d096d5c..cebf6ff1f3 100644
--- a/mindspore/ops/_op_impl/akg/gpu/squeeze.py
+++ b/mindspore/ops/_op_impl/akg/gpu/squeeze.py
@@ -15,7 +15,7 @@
 """Squeeze op"""
 from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType
 
-squeeze_op_info = AkgRegOp("SqueezeGrad") \
+squeeze_op_info = AkgRegOp("Squeeze") \
     .fusion_type("OPAQUE") \
     .input(0, "x") \
     .output(0, "output") \