aclnnEinsum

支持的产品型号

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

接口原型

每个算子分为undefined, 必须先调用aclnnEinsumGetWorkspaceSize;接口获取计算所需workspace大小以及包含了算子计算流程的执行器, 再调用aclnnEinsum接口执行计算

• aclnnStatus aclnnEinsumGetWorkspaceSize(const aclTensorList *tensors, const char * equation, aclTensor *output, uint64_t *workspaceSize, aclOpExecutor **executor)
• aclnnStatus aclnnEinsum(void *workspace, uint64_t workspaceSize, aclOpExecutor *executor, aclrtStream stream)

功能描述

• 算子功能：使用爱因斯坦求和约定对张量序列计算代数张量运算, 形式为term1, term2 -> output-term使用以下等式产生输出张量, 其中reduce-sum对出现在输入项(term1, term2)中但未出现在输出项中的所有索引执行求和。
• 计算公式： output[output-term] = reduce-sum( input1[term1] * input2[term2] )

aclnnEinsumGetWorkspaceSize

• 参数说明

  • tensors(aclTensorList*, 计算输入):Device侧的aclTensorList。支持undefined，不支持空tensor，undefined支持ND。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持FLOAT16、FLOAT、INT16、UINT16、INT32、UINT32、INT64、UINT64
    • Atlas 推理系列产品：数据类型支持FLOAT16、INT16、UINT16、INT32、UINT32、INT64、UINT64
  • equation(char*, 计算输入)：Host侧 表达式字符串。表示爱因斯坦求和约定的简写公式
  • output(aclTensor*, 计算输出)：Device侧的输出tensor。支持undefined，不支持空tensor，undefined支持ND。
    • Atlas A2训练系列产品/Atlas 800I A2推理产品：数据类型支持FLOAT16、FLOAT、INT16、UINT16、INT32、UINT32、INT64、UINT64
    • Atlas 推理系列产品：数据类型支持FLOAT16、INT16、UINT16、INT32、UINT32、INT64、UINT64
  • workspaceSize(uint64_t*, 出参)：返回需要在Device侧申请的workspace大小。
  • executor(aclOpExecutor**, 出参)：返回op执行器, 包含了算子计算流程。

• 返回值:

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

  [object Object]

aclnnEinsum

• 参数说明:

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

• 返回值:

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

约束与限制

目前equation需完全匹配, 才能找到对应函数

调用示例

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