diff --git a/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py b/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py index 31fe3322063..4e802c1cadf 100644 --- a/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py +++ b/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py @@ -24,7 +24,7 @@ hsigmoid_op_info = AkgRegOp("HSigmoid") \ .get_op_info() -@op_info_register(hsigmoidgrad_op_info) +@op_info_register(hsigmoid_op_info) def _hsigmoid_akg(): """HSigmoid AutoDiff register""" return diff --git a/mindspore/ops/_op_impl/akg/gpu/hswish.py b/mindspore/ops/_op_impl/akg/gpu/hswish.py index 17364de6cbf..29f20bafae6 100644 --- a/mindspore/ops/_op_impl/akg/gpu/hswish.py +++ b/mindspore/ops/_op_impl/akg/gpu/hswish.py @@ -24,7 +24,7 @@ hswish_op_info = AkgRegOp("HSwish") \ .get_op_info() -@op_info_register(hsigmoidgrad_op_info) +@op_info_register(hswish_op_info) def _hswish_akg(): """HSwish AutoDiff register""" return diff --git a/mindspore/ops/_op_impl/akg/gpu/hswish_grad.py b/mindspore/ops/_op_impl/akg/gpu/hswish_grad.py index 503dd9a5f1a..38e8c78e282 100644 --- a/mindspore/ops/_op_impl/akg/gpu/hswish_grad.py +++ b/mindspore/ops/_op_impl/akg/gpu/hswish_grad.py @@ -15,7 +15,7 @@ """HSwishGrad op""" from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType -hswishgrad_op_info = AkgRegOp("HSwishGrad") \ +hswish_grad_op_info = AkgRegOp("HSwishGrad") \ .fusion_type("OPAQUE") \ .input(0, "y_grad") \ .input(1, "x") \ @@ -25,7 +25,7 @@ hswishgrad_op_info = AkgRegOp("HSwishGrad") \ .get_op_info() -@op_info_register(hsigmoidgrad_op_info) +@op_info_register(hswish_grad_op_info) def _hswish_grad_akg(): """HSwishGrad AutoDiff register""" return