Asum
功能描述
计算公式
接口原型
- AspbStatus asdBlasMakeAsumPlan(asdBlasHandle handle)
- AspbStatus asdBlasSasum(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *result)
- AspbStatus asdBlasScasum(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *result)
参数列表
- asdBlasMakeAsumPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
asum算子的句柄。
- asdBlasSasum & asdBlasScasum
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
asum算子的句柄。
n
Input
const int64_t
总的元素个数。
x
Input
aclTensor *
公式中的x,Device侧的Tensor,数据格式支持ND,shape为[n]。
- Sasum下数据类型支持FLOAT32;
- Scasum下数据类型支持COMPLEX64。
incx
Input
const int64_t
相邻元素间的内存地址偏移量(当前约束为1)。
result
Output
aclTensor *
公式中的result,Device侧的Tensor,数据类型支持FLOAT32,只包含一个元素,数据格式支持ND,shape为[1]。
规格约束
- asdBlasMakeAsumPlan:无。
- asdBlasSasum
- 输入的元素个数n当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n],输出shape为[1];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
- asdBlasScasum
- 输入的元素个数n当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n],输出shape为[1];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS