UL 16_1024

Top: -

Par: 43 lines

Problem Statement

LM に ((16:1), (8_MAB:2, 8_L1B:1, 2_MAB:1, 4_PE:1, 2_W:1)) のレイアウトで分割されて配置された shape = (16, 1024) の Float Tensor \(L\) を、DRAM にある Tensor \(D\) に指定のレイアウトで格納してください。

Explanation

DL 16_1024 の反対方向の問題です。Download 操作の逆に対応した命令があるので、それを使います。

DL 16_1024 では、8 つの L2BM への放送を効率化するために、一旦 PDM を経由しましたが、今回の問題では PDM を経由せず、単に 0 番 L2BM の値を各 DRAM に転送すれば良いでしょう。

具体的な命令

• l1bmd 結合命令 [Link]
• l2bmd L1BM → L2BM 結合 [Link]
• mvp/n2048 $lc0@0.0 $d0@0 L2BM → DRAM 単独個別転送命令 [Link]

を使用します。

Inputs

• \(L\) (\(0 \le L[{i, j}] \le 2^{14}\)): Float $lm[0:32], (16,1024)/((16:1), (8_MAB:2, 8_L1B:1, 2_MAB:1, 4_PE:1, 2_W:1); B@[L2B])
  ?

Outputs

• \(D\): Float $d[0:2048]@0, (16,1024)/((4_L2B:2, 4:512), (512:1, 2_W:1))
  ?

Testcases

Submission

ログイン / 新規登録