开发者
下载
[object Object][object Object]

[object Object]

矩阵乘法 (GEMM),生成累加器/输出Tile。

[object Object]

设:

  • [object Object]
  • [object Object]
  • [object Object]

对于 [object Object][object Object](有效矩阵乘法域中的输出元素):

Ci,j=k=0K1Ai,kBk,j\mathrm{C}_{i,j} = \sum_{k=0}^{K-1} \mathrm{A}_{i,k} \cdot \mathrm{B}_{k,j}

精确的累加器行为和数据类型提升由目标/实现定义。

[object Object]

同步形式:

[object Object]
[object Object]
[object Object]
[object Object]
[object Object]
[object Object]

声明于 [object Object]

[object Object]
[object Object]
[object Object]
  • 实现检查 (Atlas A2/A3 训练系列产品/Atlas A2/A3 推理系列产品):
    • 支持的 [object Object] 三元组:
    • [object Object]
    • [object Object]
    • [object Object]
    • [object Object]
    • 静态形状约束:[object Object][object Object][object Object]
    • Tile位置:[object Object][object Object][object Object]
    • 运行时:[object Object](取自 [object Object][object Object][object Object])必须在 [object Object] 范围内。
  • 实现检查 (Ascend 950PR/Ascend 950DT):
    • 累加器类型必须是 [object Object][object Object]
    • 如果是 [object Object][object Object][object Object]
    • 如果是 [object Object]:支持 [object Object]、选定的fp8对以及 [object Object](目标定义)。
    • 静态形状约束:[object Object][object Object][object Object]
    • 强制执行分形/布局约束:
    • Left:[object Object][object Object][object Object]
    • Right:[object Object][object Object][object Object]
    • Acc:[object Object][object Object][object Object]
    • 运行时:[object Object](取自 [object Object][object Object][object Object])必须在 [object Object] 范围内。
[object Object][object Object]
[object Object]
[object Object]
[object Object]
[object Object][object Object]
[object Object]
[object Object]
[object Object]
[object Object]
[object Object]