Axpy

贡献说明

贡献者	贡献方	贡献算子	贡献时间	贡献内容
Zhoujianhua	哈尔滨工业大学-苏统华团队	Axpy	2025/12/12	新增Axpy算子

支持的产品型号

• Atlas A2训练系列产品
• Atlas 200I/500 A2推理产品

算子描述

• 功能描述

  Axpy算子计算 $y=alpha\ast 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