From 0099f0108ede55c8f723d6b69be73afedab57853 Mon Sep 17 00:00:00 2001 From: zjun Date: Tue, 14 Apr 2020 22:15:44 +0800 Subject: [PATCH] fix akg register bug --- mindspore/ops/_op_impl/akg/gpu/hsigmoid.py | 2 +- mindspore/ops/_op_impl/akg/gpu/hswish.py | 2 +- mindspore/ops/_op_impl/akg/gpu/hswish_grad.py | 4 ++-- 3 files changed, 4 insertions(+), 4 deletions(-) diff --git a/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py b/mindspore/ops/_op_impl/akg/gpu/hsigmoid.py index 31fe332206..4e802c1cad 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 17364de6cb..29f20bafae 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 503dd9a5f1..38e8c78e28 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