昇腾论坛

点赞

收藏

回复

举报

AddCustom 如何适配多种数据类型?

已解决

新人帖

发表于2024-04-11 09:59:31

0 查看

基本信息

使用Acl单算子测试
使用Atals200开发板
目前CANN8.0 alpha01
Ascend310B4

问题描述

在sample官方示例中将AddCustom->op_host->add_custom.cpp中进行修改

this->Input("x")
            .ParamType(REQUIRED)
            .DataType({ge::DT_FLOAT16, ge::DT_FLOAT})
            .Format({ge::FORMAT_ND, ge::FORMAT_ND})
            .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND});

添加了DT_FLOAT类型，但在实际进行测试，使用Acl中进行

bash run.sh

操作，此时对于半精度浮点数能够测试通过，但对于float32类型测试输出结果全为0,测试不通过，请问是只进行此处修改不可以吗？

匿名回复

我要发帖子