From 80aadc10ed016d6d4656bc13e03f678be30a1311 Mon Sep 17 00:00:00 2001 From: admin Date: Sun, 23 Aug 2026 10:21:14 +0000 Subject: [PATCH] =?UTF-8?q?PPT=E4=BF=AE=E6=AD=A3=20StreamK=20=E6=9D=A1?= =?UTF-8?q?=E4=BB=B6=E4=B8=BA=20P

9. StreamK — K 维核间切分

思路:B/M/N 都填不满核时,把 K 切给多个核各算一段,再归约

-

进入条件:P = B·MN·4B/L0C < 32;K/grid_K ≥ 128 元素(dValue 256B)

+

进入条件:P = B·MN·4B/L0C < 16(=C/2);K/grid_K ≥ 128 元素(dValue 256B)

+

阈值取 C/2:grid_K≥2 时每块需 2 核,P×2 ≤ C 才放得下

归约代价(按实现流程推导):

第1步:grid_K 个 AIC 各算 K/grid_K 段,L0C 原地累加 第2步:fixpipe 写部分和到 workspace(驻留 L2,走 5.2TB/s 写口)