aclnnRound&aclnnInplaceRound
支持的产品型号
- [object Object]Atlas 训练系列产品[object Object]。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]。
函数原型
aclnnRound和aclnnInplaceRound实现相同的功能,使用区别如下,请根据自身实际场景选择合适的算子。
- aclnnRound:需新建一个输出张量对象存储计算结果。
- aclnnInplaceRound:无需新建输出张量对象,直接在输入张量的内存中存储计算结果。
每个算子分为undefined,必须先调用“aclnnRoundGetWorkspaceSize”或者”aclnnInplaceRoundGetWorkspaceSize“接口获取入参并根据流程计算所需workspace大小,再调用“aclnnRound”或者”aclnnInplaceRound“接口执行计算。
aclnnStatus aclnnRoundGetWorkspaceSize(const aclTensor *self, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
aclnnStatus aclnnRound(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, const aclrtStream stream)
aclnnStatus aclnnInplaceRoundGetWorkspaceSize(const aclTensor *selfRef, uint64_t *workspaceSize, aclOpExecutor **executor)
aclnnStatus aclnnInplaceRound(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, const aclrtStream stream)
功能说明
算子功能:将输入的值舍入到最接近的整数,若该值与两个整数距离一样则向偶数取整。
aclnnRoundGetWorkspaceSize
参数说明:
- self(aclTensor*,计算输入):Device侧的aclTensor,undefined支持ND,shape不支持9D及以上。
- [object Object]Atlas 训练系列产品[object Object]:数据类型支持FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]:数据类型支持BFLOAT16、FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- out(aclTensor *,计算输出):Device侧的aclTensor,shape与self一致,dtype与self一致,undefined支持ND。
- [object Object]Atlas 训练系列产品[object Object]:数据类型支持FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]:数据类型支持BFLOAT16、FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- workspaceSize(uint64_t*, 出参):返回需要在Device侧申请的workspace大小。
- executor(aclOpExecutor**, 出参):返回op执行器,包含了算子计算流程。
- self(aclTensor*,计算输入):Device侧的aclTensor,undefined支持ND,shape不支持9D及以上。
返回值:
aclnnStatus: 返回状态码,具体参见undefined。
aclnnRound
参数说明:
- workspace(void*, 入参):在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参): 在Device侧申请的workspace大小, 由第一段接口aclnnRoundGetWorkspaceSize获取。
- executor(aclOpExecutor*, 入参): op执行器, 包含了算子计算流程。
- stream(aclrtStream, 入参): 指定执行任务的AscendCL Stream流。
返回值:
aclnnStatus: 返回状态码,具体参见undefined。
aclnnInplaceRoundGetWorkspaceSize
参数说明:
- selfRef(aclTensor*,计算输入):Device侧的aclTensor,undefined支持ND,shape不支持9D及以上。
- [object Object]Atlas 训练系列产品[object Object]:数据类型支持FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- [object Object]Atlas A2 训练系列产品/Atlas 800I A2 推理产品/A200I A2 Box 异构组件[object Object]:数据类型支持BFLOAT16、FLOAT16、 FLOAT32、 DOUBLE、 INT32、 INT64。
- workspaceSize(uint64_t*, 出参):返回需要在Device侧申请的workspace大小。
- executor(aclOpExecutor**, 出参):返回op执行器,包含了算子计算流程。
- selfRef(aclTensor*,计算输入):Device侧的aclTensor,undefined支持ND,shape不支持9D及以上。
返回值:
aclnnStatus: 返回状态码,具体参见undefined。
aclnnInplaceRound
参数说明:
- workspace(void*, 入参):在Device侧申请的workspace内存地址。
- workspaceSize(uint64_t, 入参):在Device侧申请的workspace大小, 由第一段接口aclnnInplaceRoundGetWorkspaceSize获取。
- executor(aclOpExecutor*, 入参):op执行器, 包含了算子计算流程。
- stream(aclrtStream, 入参):指定执行任务的AscendCL Stream流。
返回值:
aclnnStatus: 返回状态码,具体参见undefined。
约束说明
- 当输入值在[-0.5, -0]之间时,输出值为0
调用示例
示例代码如下,仅供参考,具体编译和执行过程请参考undefined。