aclCreateFloatArray
函数功能
创建aclFloatArray对象,作为单算子API执行接口的入参。
aclFloatArray是AscendCL定义的一种用来管理和存储浮点型数据的数组结构,开发者无需关注其内部实现,直接使用即可。
函数原型
aclFloatArray *aclCreateFloatArray(const float *value, uint64_t size)
参数说明
参数名 |
输入/输出 |
说明 |
---|---|---|
value |
输入 |
Host侧的float类型指针,其指向的值会拷贝给aclFloatArray。 |
size |
输入 |
浮点型数组的长度,取值大于0。 |
返回值说明
成功则返回创建好的aclFloatArray,否则返回nullptr。
约束与限制
- 本接口需与aclDestroyFloatArray接口配套使用,分别完成aclFloatArray的创建与销毁。
- 调用aclGetFloatArraySize接口可以获取aclFloatArray的大小。
调用示例
关键代码示例如下,仅供参考,不支持直接拷贝运行。
1 2 3 4 5 6 7 8 9 10 | // 创建aclFloatArray std::vector<float> scalesData = {1.0, 1.0, 2.0, 2.0}; aclFloatArray *scales = aclCreateFloatArray(scalesData.data(),scalesData.size()); ... // aclFloatArray作为单算子API执行接口的入参 auto ret = aclnnXxxGetWorkspaceSize(srcTensor, scales, ..., outTensor, ..., &workspaceSize, &executor); ret = aclnnXxx(...); ... // 销毁aclFloatArray ret = aclDestroyFloatArray(scales); |
父主题: 公共接口