昇腾社区首页
中文
注册

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执行器,包含了算子计算流程。
  • 返回值:

    aclnnStatus: 返回状态码,具体参见undefined

[object Object]

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执行器,包含了算子计算流程。
  • 返回值:

    aclnnStatus: 返回状态码,具体参见undefined

[object Object]

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

[object Object]