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") \