Cal
产品支持情况
硬件型号 |
支持情况 |
---|---|
|
不支持 |
|
不支持 |
|
不支持 |
|
支持 |
|
支持 |
功能说明
接口功能:
- asdBlasMakeCalPlan:初始化该句柄对应的Cal算子配置。
- asdBlasSscal:对一个实数向量进行缩放,即将向量中的每个元素乘以一个实数alpha。
- asdBlasCsscal:对一个复数向量进行缩放,即将向量中的每个元素乘以一个实数alpha。
- asdBlasCscal:对一个复数向量进行缩放,即将向量中的每个元素乘以一个复数alpha。
计算公式
- asdBlasSscal的计算公式:
- asdBlasCsscal的计算公式:
- asdBlasCscal的计算公式:
函数原型
- AspbStatus asdBlasMakeCalPlan(asdBlasHandle handle)
- AspbStatus asdBlasSscal(asdBlasHandle handle, const int64_t n, const float *alpha, aclTensor *x, const int64_t incx)
- AspbStatus asdBlasCsscal(asdBlasHandle handle, const int64_t n, const float *alpha, aclTensor *x, const int64_t incx)
- AspbStatus asdBlasCscal(asdBlasHandle handle, const int64_t n, const std::complex<float> *alpha, aclTensor *x,const int64_t incx)
参数说明
- asdBlasMakeCalPlan
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
cal算子的句柄。
- asdBlasSscal & asdBlasCsscal & asdBlasCscal
参数名称
Input/Output
类型
描述
handle
Input
asdBlasHandle
sscal算子的句柄。
n
Input
const int64_t
总的元素个数。
alpha
Input
const float *
公式中的alpha,向量的缩放因子。
x
Input/Output
aclTensor *
公式中的x,Device侧的Tensor,数据类型支持FLOAT32,数据格式支持ND,shape为[n]。
incx
Input
const int64_t
相邻元素间的内存地址偏移量(当前约束为1)。
约束说明
- asdBlasMakeCalPlan:无。
- asdBlasSscal
- 输入的元素个数当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
- asdBlasCsscal
- 输入的元素个数当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)。
- asdBlasCscal
- 输入的元素个数当前覆盖支持[1,6.71e+06];
- 算子输入shape为[n],输出shape为[n];
- 算子实际计算时,不支持ND高维度运算(不支持维度≥3的运算)
调用示例
算子的调用示例参见Cal。
父主题: BLAS