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