Axpy
贡献说明
| 贡献者 | 贡献方 | 贡献算子 | 贡献时间 | 贡献内容 |
|---|---|---|---|---|
| Zhoujianhua | 哈尔滨工业大学-苏统华团队 | Axpy | 2025/12/12 | 新增Axpy算子 |
支持的产品型号
- Atlas A2训练系列产品
- Atlas 200I/500 A2推理产品
算子描述
-
功能描述
Axpy算子计算 y=alpha∗x+yy = alpha * x + y。 -
原型信息
算子类型(OpType) Axpy name Type data type format 算子输入 x tensor float32,float16 ND y tensor float32,float16 ND alpha a float32 ND 算子输出 z tensor float32,float16 ND 核函数名 axpy
约束与限制
- x,y,out的数据类型仅支持float32,float16,数据格式仅支持ND
算子使用
使用该算子前,请参考社区版CANN开发套件包安装文档完成开发运行环境的部署。
编译部署
-
进入到仓库目录
cd ${git_clone_path}/ops-math -
执行编译
bash build.sh --experimental --ops=axpy_v3 --soc=ascend910b --pkg -
部署算子包
./build_out/cann-ops-math-custom_linux-aarch64.run
算子调用
-
执行调用
bash build.sh --run_example axpy_v3 eager cust --vendor_name=custom
调用说明
| 调用方式 | 调用样例 | 说明 |
|---|---|---|
| aclnn调用 | test_aclnn_axpy |