下载
EN
注册

aclCreateScalar

函数功能

创建aclScalar,作为CANN算子API接口的入参。

aclScalar由AscendCL框架定义,用来管理和存储标量数据(比如一个单独的数),开发者无需关注其内部实现。

函数原型

aclScalar *aclCreateScalar(void *value, aclDataType dataType)

参数说明

aclDataType定义由AscendCL提供,具体参见应用开发接口aclDataType

参数名

输入/输出

说明

value

输入

Host侧的scalar类型的指针,其指向的值会作为scalar。

dataType

输入

scalar的数据类型。

返回值说明

成功则返回创建好的aclScalar,否则返回nullptr。

约束与限制

本接口与aclDestroyScalar接口配套使用,分别完成aclScalar的创建与销毁。

调用示例

关键代码示例如下,不可以直接拷贝运行,仅供参考。

// 创建aclScalar
float alphaValue = 1.2f;
aclScalar* alpha = aclCreateScalar(&alphaValue, aclDataType::ACL_FLOAT);
...
// aclScalar作为单算子API执行接口的入参
auto ret = aclnnXxxGetWorkspaceSize(srcTensor, alpha, ..., outTensor, ..., &workspaceSize, &executor);
ret = aclnnXxx(...);
...
// 销毁aclScalar
ret = aclDestroyScalar(alpha);