aclnnAddcdiv&aclnnInplaceAddcdiv

支持的产品型号

• Atlas 推理系列产品。
• Atlas 训练系列产品。
• Atlas A2训练系列产品/Atlas 800I A2推理产品。

接口原型

• aclnnAddcdiv和aclnnInplaceAddcdiv实现相同的功能，使用区别如下，请根据自身实际场景选择合适的算子。

  • aclnnAddcdiv：需新建一个输出张量对象存储计算结果。
  • aclnnInplaceAddcdiv：无需新建输出张量对象，直接在输入张量的内存中存储计算结果。

• 每个算子分为undefined，必须先调用 “aclnnAddcdivGetWorkspaceSize” 或者 “aclnnInplaceAddcdivGetWorkspaceSize” 接口获取入参并根据计算流程计算所需workspace大小，再调用 “aclnnAddcdiv” 或者 “aclnnInplaceAddcdiv” 接口执行计算。

  • aclnnStatus aclnnAddcdivGetWorkspaceSize(const aclTensor* self, const aclTensor* tensor1, const aclTensor* tensor2, const aclScalar* value, aclTensor* out, uint64_t* workspaceSize, aclOpExecutor** executor)
  • aclnnStatus aclnnAddcdiv(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, const aclrtStream stream)
  • aclnnStatus aclnnInplaceAddcdivGetWorkspaceSize(const aclTensor* selfRef, const aclTensor* tensor1, const aclTensor* tensor2, const aclScalar* value, uint64_t* workspaceSize, aclOpExecutor** executor)
  • aclnnStatus aclnnInplaceAddcdiv(void* workspace, uint64_t workspaceSize, aclOpExecutor* executor, const aclrtStream stream)

功能描述

• 算子功能：执行 tensor1 除以 tensor2 的元素除法，将结果乘以标量 value 并将其添加到 self 。

• 计算公式：

$$out_i = self_i + value \times {tensor1_i \over tensor2_i}$$

aclnnAddcdivGetWorkspaceSize

• 参数说明：

  • self（aclTensor*, 计算输入）：公式中的输入self，Device侧的aclTensor，self与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），self与tensor1、tensor2的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • tensor1（aclTensor*, 计算输入）：公式中的输入tensor1，Device侧的aclTensor，self与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），self与tensor1、tensor2的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • tensor2（aclTensor*, 计算输入）：公式中的输入tensor2，Device侧的aclTensor，self与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），self与tensor1、tensor2的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • value（aclScalar*, 计算输入）：公式中的输入value，host侧的aclScalar，数据类型需要可转换成self与tensor1、tensor2推导后的数据类型（参见undefined）。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • out（aclTensor*, 计算输出）：公式中的输出out，Device侧的aclTensor，且数据类型是self与tensor1、tensor2推导之后可转换的数据类型（参见undefined），out与self、tensor1、tensor2 broadcast之后的tensor的shape一致，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • workspaceSize（uint64_t*, 出参）：返回需要在Device侧申请的workspace大小。
  • executor（aclOpExecutor**, 出参）：返回op执行器，包含了算子计算流程。

• 返回值：

  aclnnStatus：返回状态码，具体参见undefined。

  [object Object]

aclnnAddcdiv

• 参数说明：

  • workspace（void*, 入参）：在Device侧申请的workspace内存地址。
  • workspaceSize（uint64_t, 入参）：在Device侧申请的workspace大小，由第一段接口aclnnAddcdivGetWorkspaceSize获取。
  • executor（aclOpExecutor*, 入参）：op执行器，包含了算子计算流程。
  • stream（aclrtStream, 入参）：指定执行任务的AscendCL Stream流。

• 返回值：

  aclnnStatus：返回状态码，具体参见undefined。

aclnnInplaceAddcdivGetWorkspaceSize

• 参数说明：

  • selfRef（aclTensor*, 计算输入/输出）：公式中的输入self和输出out，Device侧的aclTensor，selfRef与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），且推导后的数据类型可以转换为selfRef的数据类型（参见undefined），selfRef与tensor1和tensor2 broadcast之后的tensor的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • tensor1（aclTensor*, 计算输入）：公式中的输入tensor1，Device侧的aclTensor，selfRef与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），tensor1与tensor2的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • tensor2（aclTensor*, 计算输入）：公式中的输入tensor2，Device侧的aclTensor，selfRef与tensor1、tensor2的数据类型满足数据类型推导规则（参见undefined），tensor1与tensor2的shape满足undefined，支持undefined，undefined支持ND。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • value（aclScalar*, 计算输入）：公式中的输入value，host侧的aclScalar，数据类型需要可转换成selfRef与tensor1、tensor2推导后的数据类型（参见undefined）。
    • Atlas 训练系列产品：数据类型支持FLOAT16、FLOAT、DOUBLE、INT64。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持BFLOAT16、FLOAT16、FLOAT、DOUBLE、INT64。
  • workspaceSize（uint64_t*, 出参）：返回需要在Device侧申请的workspace大小。
  • executor（aclOpExecutor**, 出参）：返回op执行器，包含了算子计算流程。

• 返回值：

  aclnnStatus：返回状态码，具体参见undefined。

  [object Object]

aclnnInplaceAddcdiv

• 参数说明：

  • workspace（void*, 入参）：在Device侧申请的workspace内存地址。
  • workspaceSize（uint64_t, 入参）：在Device侧申请的workspace大小，由第一段接口aclnnInplaceAddcdivGetWorkspaceSize获取。
  • executor（aclOpExecutor*, 入参）：op执行器，包含了算子计算流程。
  • stream（aclrtStream, 入参）：指定执行任务的AscendCL Stream流。

• 返回值：

  aclnnStatus：返回状态码，具体参见undefined。

约束与限制

无

调用示例

示例代码如下，仅供参考，具体编译和执行过程请参考undefined。 aclnnAddcdiv

aclnnInplaceAddcdiv