PRLu算子UT测试问题

最新推荐文章于 2024-02-28 11:24:30 发布

xiao | yang

最新推荐文章于 2024-02-28 11:24:30 发布

阅读量225

点赞数

分类专栏：深度学习文章标签： python 开发语言

本文链接：https://blog.csdn.net/xi_xiyu/article/details/127645843

版权

在进行MindSpore自定义PReLU算子的单元测试(UT)时遇到错误，测试报告指出可能的weight shape问题。文档中提到weight需与输入x通道数相同，对于GPU设备，若输入为标量，shape应为1。开发环境为DSL和MindSpore框架。错误信息'Get realpath of input failed'提示输入路径不正确，建议检查输入的shape、type和value。

摘要由CSDN通过智能技术生成

报错：

测试报告：http://localhost:63342/MyOperatorp/out/coverage_report/ut_impl/d_c585556f9f416c9f_p_re_lu_impl_py.html

测试代码：

from op_test_frame.ut import OpUT
from mindspore import nn
# [TODO] change op_func_name here
ut_case = OpUT("PReLU", "impl.p_re_lu_impl", "p_re_lu_impl")
def calc_expect_func(x, y, weight):
    res = nn.PReLU(x.get("value"),weight.get("value"))
    return [res, ]


ut_case.add_precision_case("all", {
    "params": [
        {"dtype": "float16", "format": "NCHW", "ori_format": "NCHW", "ori_shape": (1,1,1), "shape": (1,1,1),
         "param_type": "input"},
        {"dtype": "float16", "format": "NCHW", "ori_format": &#