aclnnLog1p&aclnnInplaceLog1p

📄 查看源码

产品支持情况

产品 是否支持
Ascend 950PR/Ascend 950DT
Atlas A3 训练系列产品/Atlas A3 推理系列产品
Atlas A2 训练系列产品/Atlas A2 推理系列产品
Atlas 200I/500 A2 推理产品 ×
Atlas 推理系列产品
Atlas 训练系列产品

功能说明

  • 算子功能:对输入Tensor完成log1p运算
  • 计算公式:

out= log1p(self)= loge(self+1)out = {\ {{log1p}{(self)}}} = {\ {{log_e}{(self+1)}}}

函数原型

  • aclnnLog1p和aclnnInplaceLog1p实现相同的功能,使用区别如下,请根据自身实际场景选择合适的算子。
    • aclnnLog1p:需新建一个输出张量对象存储计算结果。
    • aclnnInplaceLog1p:无需新建输出张量对象,直接在输入张量的内存中存储计算结果。
  • 每个算子分为两段式接口,必须先调用“aclnnLog1pGetWorkspaceSize”或者“aclnnInplaceLog1pGetWorkspaceSize”接口获取计算所需workspace大小以及包含了算子计算流程的执行器,再调用“aclnnLog1p”或者“aclnnInplaceLog1p”接口执行计算。
    • aclnnStatus aclnnLog1pGetWorkspaceSize(const aclTensor *self, aclTensor *out, uint64_t *workspaceSize, aclOpExecutor **executor)
    • aclnnStatus aclnnLog1p(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)
    • aclnnStatus aclnnInplaceLog1pGetWorkspaceSize(aclTensor* selfRef, uint64_t* workspaceSize, aclOpExecutor** executor)
    • aclnnStatus aclnnInplaceLog1p(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

aclnnLog1pGetWorkspaceSize

  • 参数说明

    • self(aclTensor*,计算输入): 公式中的self,Device侧的aclTensor。支持非连续的Tensor数据格式支持ND,shape维度不大于8, 且shape需要与out一致,和out的数据满足数据类型推导规则。
      • Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持INT8、INT16、INT32、INT64、UINT8、BOOL、FLOAT、FLOAT16、DOUBLE。
      • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持INT8、INT16、INT32、INT64、UINT8、BOOL、FLOAT、FLOAT16、DOUBLE、BFLOAT16。
    • out(aclTensor *,计算输出): 公式中的out,Device侧的aclTensor。支持非连续的Tensor数据格式支持ND,且shape需要与self一致。
      • Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持FLOAT、FLOAT16、DOUBLE。
      • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持FLOAT、FLOAT16、DOUBLE、BFLOAT16。
    • workspaceSize(uint64_t *,出参):返回需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor **,出参):返回op执行器,包含了算子计算流程。
  • 返回值:

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

第一段接口完成入参校验,出现以下场景时报错:
161001(ACLNN_ERR_PARAM_NULLPTR):1. 传入的self或out是空指针。
161002(ACLNN_ERR_PARAM_INVALID):1. self和out的数据类型和数据格式不在支持的范围之内。
                                 2. self和out的数据类型不符合数据类型推导规则。
                                 3. Self和Out的维度大于8。
                                 4. Self和Out的shape不一致。

aclnnLog1p

  • 参数说明

    • workspace(void *,入参):在Device侧申请的workspace内存地址。
    • workspaceSize(uint64_t,入参):在Device侧申请的workspace大小,由第一段接口aclnnLog1pGetWorkspaceSize获取。
    • executor(aclOpExecutor *,入参):op执行器,包含了算子计算流程。
    • stream(aclrtStream,入参):指定执行任务的Stream。
  • 返回值

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

aclnnInplaceLog1pGetWorkspaceSize

  • 参数说明

    • selfRef(aclTensor *,计算输入|计算输出): Device侧的aclTensor。支持非连续的Tensor数据格式支持ND,shape维度不大于8。
      • Atlas 推理系列产品、Atlas 训练系列产品:数据类型支持FLOAT、FLOAT16、DOUBLE。
      • Atlas A2 训练系列产品/Atlas A2 推理系列产品、Atlas A3 训练系列产品/Atlas A3 推理系列产品、Ascend 950PR/Ascend 950DT:数据类型支持FLOAT、FLOAT16、DOUBLE、BFLOAT16。
    • workspaceSize(uint64_t *,出参):返回需要在Device侧申请的workspace大小。
    • executor(aclOpExecutor **,出参):返回op执行器,包含了算子计算流程。
  • 返回值

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

第一段接口完成入参校验,出现以下场景时报错:
161001(ACLNN_ERR_PARAM_NULLPTR):1. 传入的selfRef是空指针。
161002(ACLNN_ERR_PARAM_INVALID):1. selfRef数据类型和数据格式不在支持的范围之内。
                                 2. selfRef的维度大于8。

aclnnInplaceLog1p

  • 参数说明

    • workspace(void *,入参):在Device侧申请的workspace内存地址。
    • workspaceSize(uint64_t,入参):在Device侧申请的workspace大小,由第一段接口aclnnInplaceLog1pGetWorkspaceSize获取。
    • executor(aclOpExecutor *,入参):op执行器,包含了算子计算流程。
    • stream(aclrtStream,入参):指定执行任务的Stream。
  • 返回值

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

约束说明

  • 确定性计算:
    • aclnnLog1p&aclnnInplaceLog1p默认确定性实现。