aclnnEinsum

支持的产品型号

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

接口原型

每个算子分为, 必须先调用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, 数据类型支持FLOAT16、FLOAT（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）、INT16、UINT16、INT32、UINT32、INT64、UINT64 。支持， 支持ND。
  • equation(const char*, 计算输入): Host侧 表达式字符串。表示爱因斯坦求和约定的简写公式
  • output(aclTensor*, 计算输出): Device侧的输出tensor,数据类型支持FLOAT16、FLOAT（仅Atlas A2训练系列产品/Atlas 800I A2推理产品支持）、INT16、UINT16、INT32、UINT32、INT64、UINT64。支持ND。
  • workspaceSize(uint64_t*, 计算输出):返回用户需要在Device侧申请的workspace大小。
  • executor(aclOpExecutor**, 计算输出):返回op执行器, 包含了算子计算流程。

• 返回值:

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

  [object Object]

aclnnEinsum

• 参数说明:

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

• 返回值:

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

约束与限制

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

调用示例

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