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