aclCreateScalar
函数功能
创建aclScalar,作为CANN算子API接口的入参。
aclScalar由AscendCL框架定义,用来管理和存储标量数据(比如一个单独的数),开发者无需关注其内部实现。
函数原型
aclScalar *aclCreateScalar(void *value, aclDataType dataType)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
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);
父主题: 基础接口