EluGradV2算子精度是否有问题
收藏回复举报
EluGradV2算子精度是否有问题
t('forum.solved') 已解决
发表于2024-06-11 12:50:27
0 查看

【用户类型】:企业

【昇腾产品型号】:910A

【版本信息】:

  • 操作系统版本:Ubuntu 20.04
  • 固件驱动版本:npu-smi 23.0.0 Version: 23.0.0
  • CANN版本:CANN 7.0.1

【官方样例or个人代码】:

【开发内容】:适配昇腾算子

【问题描述】:使用aclopCompileAndExecute执行EluGradV2算子,精度和pytorch对比,发现Tensor中有些值差距较大。

复现方法

克隆仓库

git clone cid:link_0

生成数据

cd EluGrad
python elu_grad.py

这一步会将x、out、dout、dx都存入文件中

执行算子

cd ..
bash run_demo.sh EluGrad

算子将从out.bin、dout.bin中读取数据作为EluGradV2算子的输入,计算出的结果与dx.bin文件中进行对比。

我要发帖子