公共接口
本节罗列了单算子API执行依赖的一些AscendCL元接口,通过这些元接口可以构建所需的数据结构,如aclTensor、aclScalar、aclIntArray等。
接口名 |
接口定义 |
功能说明 |
参数说明 |
返回值 |
---|---|---|---|---|
aclCreateTensor |
aclTensor *aclCreateTensor(const int64_t *viewDims, uint64_t viewDimsNum, aclDataType dataType, const int64_t *stride, int64_t offset, aclFormat format, const int64_t *storageDims, uint64_t storageDimsNum, void *tensorData) |
创建aclTensor。 |
|
成功则返回创建好的aclTensor,否则返回nullptr。 |
aclDestroyTensor |
aclnnStatus aclDestroyTensor(const aclTensor *tensor) |
销毁aclTensor。 |
tensor(输入):需要销毁的tensor指针。 |
返回0表示成功,返回其它值表示失败。 |
aclCreateScalar |
aclScalar *aclCreateScalar(void *value, aclDataType dataType) |
创建aclScalar。 |
|
成功则返回创建好的aclScalar,否则返回nullptr。 |
aclDestroyScalar |
aclnnStatus aclDestroyScalar(const aclScalar *scalar) |
销毁aclScalar。 |
scalar(输入):需要销毁的scalar。 |
返回0表示成功,返回其它值表示失败。 |
aclCreateIntArray |
aclIntArray *aclCreateIntArray(const int64_t *value, uint64_t size) |
创建aclIntArray。 |
|
成功则返回创建好的aclIntArray,否则返回nullptr。 |
aclDestroyIntArray |
aclnnStatus aclDestroyIntArray(const aclIntArray *array) |
销毁aclIntArray。 |
array(输入):需要销毁的aclIntArray。 |
返回0表示成功,返回其它值表示失败。 |
aclCreateFloatArray |
aclFloatArray *aclCreateFloatArray(const float *value, uint64_t size) |
创建aclFloatArray。 |
|
成功则返回创建好的aclFloatArray,否则返回nullptr。 |
aclDestroyFloatArray |
aclnnStatus aclDestroyFloatArray(const aclFloatArray *array) |
销毁aclFloatArray。 |
array(输入):需要销毁的aclFloatArray |
返回0表示成功,返回其它值表示失败。 |
aclCreateBoolArray |
aclBoolArray *aclCreateBoolArray(const bool *value, uint64_t size) |
创建aclBoolArray。 |
|
成功则返回创建好的aclBoolArray,否则返回nullptr。 |
aclDestroyBoolArray |
aclnnStatus aclDestroyBoolArray(const aclBoolArray *array) |
销毁aclBoolArray。 |
array(输入):需要销毁的aclBoolArray。 |
返回0表示成功,返回其它值表示失败。 |
aclCreateTensorList |
aclTensorList *aclCreateTensorList(const aclTensor *const *value, uint64_t size) |
创建aclTensorList。 |
value(输入): Device侧的aclTensor类型的指针,其指向的值会赋给TensorList。 size(输入):张量列表的长度。 |
成功则返回创建好的aclTensorList,否则返回nullptr。 |
aclDestroyTensorList |
aclnnStatus aclDestroyTensorList(const aclTensorList *array) |
销毁aclTensorList。 |
array(输入):需要销毁的aclTensorList。 |
返回0表示成功,返回其它值表示失败。 |
aclGetViewShape |
aclnnStatus aclGetViewShape(const aclTensor *tensor, int64_t **viewDims, uint64_t *viewDimsNum) |
获取aclTensor的ViewShape。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetViewStrides |
aclnnStatus aclGetViewStrides(const aclTensor *tensor, int64_t **stridesValue, uint64_t *stridesNum) |
获取aclTensor的viewStrides。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetViewOffset |
aclnnStatus aclGetViewOffset(const aclTensor *tensor, int64_t *offset) |
获取aclTensor的viewOffset。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetFormat |
aclnnStatus aclGetFormat(const aclTensor *tensor, aclFormat *format) |
获取aclTensor的format。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetDataType |
aclnnStatus aclGetDataType(const aclTensor *tensor, aclDataType *dataType) |
获取aclTensor的DataType。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetIntArraySize |
aclnnStatus aclGetIntArraySize(const aclIntArray *array, uint64_t *size) |
获取aclIntArray的大小。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetFloatArraySize |
aclnnStatus aclGetFloatArraySize(const aclFloatArray *array, uint64_t *size) |
获取aclFloatArray的大小。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetBoolArraySize |
aclnnStatus aclGetBoolArraySize(const aclBoolArray *array, uint64_t *size) |
获取aclBoolArray的大小。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetTensorListSize |
aclnnStatus aclGetTensorListSize(const aclTensorList *tensorList, uint64_t *size) |
获取aclTensorList的大小。 |
|
返回0表示成功,返回其它值表示失败。 |
aclGetStorageShape |
aclnnStatus aclGetStorageShape(const aclTensor *tensor, int64_t **storageDims, uint64_t *storageDimsNum) |
或取Tensor在内存上的存储shape。 |
|
返回0表示成功,返回其它值表示失败。 |
aclCreateScalarList |
aclScalarList *aclCreateScalarList(const aclScalar *const *value, uint64_t size) |
创建aclScalarList。 |
value(输入): Device侧的aclScalar类型的指针,其指向的值会赋给ScalarList。 size(输入):标量列表的长度。 |
成功则返回创建好的aclScalarList,否则返回nullptr。 |
aclDestroyScalarList |
aclnnStatus aclDestroyScalarList(const aclScalarList *array) |
销毁aclScalarList。 |
array(输入):需要销毁的aclScalarList。 |
返回0表示成功,返回其它值表示失败。 |
aclGetScalarListSize |
aclnnStatus aclGetScalarListSize(const aclScalarList *scalarList, uint64_t *size) |
获取aclScalarList的大小。 |
|
返回0表示成功,返回其它值表示失败。 |
aclInitTensor |
aclnnStatus aclInitTensor(aclTensor *tensor, const int64_t *viewDims, uint64_t viewDimsNum, aclDataType dataType, const int64_t *stride, int64_t offset, aclFormat format, const int64_t *storageDims, uint64_t storageDimsNum, void *tensorDataAddr) |
初始化给定tensor的参数。 当用户想复用已创建aclTensor时,可使用该接口先重置aclTensor的各项属性。 |
|
返回0表示成功,返回其它值表示失败。 |