Csrot
计算公式
接口原型
- AspbStatus asdBlasMakeRotPlan(asdBlasHandle handle)
- AspbStatus asdBlasCsrot(asdBlasHandle handle, const int64_t n, aclTensor *x, const int64_t incx, aclTensor *y,const int64_t incy, const float *c, const float *s)
参数列表
- asdBlasMakeRotPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
csrot算子的句柄。
- asdBlasCsrot
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
csrot算子的句柄。
n
Input
const int64_t
向量中复数元素的个数。
x
Input/Output
aclTensor *
公式中的x,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。
incx
Input
const int64_t
x相邻元素间的内存地址偏移量(当前约束为1)。
y
Input/Output
aclTensor *
公式中的y,Device侧的Tensor,数据类型支持COMPLEX64,数据格式支持ND,shape为[n]。
incy
Input
const int64_t
y相邻元素间的内存地址偏移量(当前约束为1)。
c
Input
const float *
旋转矩阵的余弦值指针。
s
Input
const float *
旋转矩阵的正弦值指针。
规格约束
- asdBlasMakeRotPlan:无。
- asdBlasCsrot
- 输入的元素个数n当前覆盖支持[1,2.50e+06];
- 算子输入shape为[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
父主题: BLAS