Caxpy
计算公式
接口原型
- AspbStatus asdBlasMakeCaxpyPlan(asdBlasHandle handle)
- AspbStatus asdBlasCaxpy(asdBlasHandle handle, const int64_t n, const std::complex<float> *alpha, aclTensor* x,int64_t incx, aclTensor* y, int64_t incy)
参数列表
- asdBlasMakeCaxpyPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
Caxpy算子的句柄。
- asdBlasCaxpy
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
Caxpy算子的句柄。
n
Input
const int64_t
向量中复数元素的个数。
alpha
Input
const std::complex<float> *
公式中的alpha,输入的复数标量,数据类型支持COMPLEX64。
x
Input
aclTensor*
公式中的x,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。
incx
Input
int64_t
向量x相邻元素间的内存地址偏移量(当前约束为1)。
y
Input/Output
aclTensor*
公式中的y,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。
incy
Input
int64_t
向量y相邻元素间的内存地址偏移量(当前约束为1)。
规格约束
- asdBlasMakeCaxpyPlan:无。
- asdBlasCaxpy
- 输入的元素个数n当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n]、[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS