p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix akg register bug

This commit is contained in:
zjun 2020-04-14 22:15:44 +08:00
parent d90e121547
commit 0099f0108e
3 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ops/_op_impl/akg/gpu/hsigmoid.py
View File

@ -24,7 +24,7 @@ hsigmoid_op_info = AkgRegOp("HSigmoid") \
.get_op_info() .get_op_info()
@op_info_register(hsigmoidgrad_op_info) @op_info_register(hsigmoid_op_info)
def _hsigmoid_akg(): def _hsigmoid_akg():
"""HSigmoid AutoDiff register""" """HSigmoid AutoDiff register"""
return return

2
mindspore/ops/_op_impl/akg/gpu/hswish.py
View File

@ -24,7 +24,7 @@ hswish_op_info = AkgRegOp("HSwish") \
.get_op_info() .get_op_info()
@op_info_register(hsigmoidgrad_op_info) @op_info_register(hswish_op_info)
def _hswish_akg(): def _hswish_akg():
"""HSwish AutoDiff register""" """HSwish AutoDiff register"""
return return

4
mindspore/ops/_op_impl/akg/gpu/hswish_grad.py
View File

@ -15,7 +15,7 @@
"""HSwishGrad op""" """HSwishGrad op"""
from mindspore.ops.op_info_register import op_info_register, AkgRegOp, DataType 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") \ .fusion_type("OPAQUE") \
.input(0, "y_grad") \ .input(0, "y_grad") \
.input(1, "x") \ .input(1, "x") \
@ -25,7 +25,7 @@ hswishgrad_op_info = AkgRegOp("HSwishGrad") \
.get_op_info() .get_op_info()
@op_info_register(hsigmoidgrad_op_info) @op_info_register(hswish_grad_op_info)
def _hswish_grad_akg(): def _hswish_grad_akg():
"""HSwishGrad AutoDiff register""" """HSwishGrad AutoDiff register"""
return return