Browse Source

fix akg register bug

tags/v0.2.0-alpha
zjun 5 years ago
parent
commit
0099f0108e
3 changed files with 4 additions and 4 deletions
  1. +1
    -1
      mindspore/ops/_op_impl/akg/gpu/hsigmoid.py
  2. +1
    -1
      mindspore/ops/_op_impl/akg/gpu/hswish.py
  3. +2
    -2
      mindspore/ops/_op_impl/akg/gpu/hswish_grad.py

+ 1
- 1
mindspore/ops/_op_impl/akg/gpu/hsigmoid.py View File

@@ -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

+ 1
- 1
mindspore/ops/_op_impl/akg/gpu/hswish.py View File

@@ -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

+ 2
- 2
mindspore/ops/_op_impl/akg/gpu/hswish_grad.py View File

@@ -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

Loading…
Cancel
Save