Broadcast - 动态 UB Broadcast（DAV_3510）

【免费下载链接】cannbot-skills CANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。 项目地址: https://gitcode.com/cann/cannbot-skills

适用场景: 合轴后多维，DAV_3510 芯片。使用动态 Broadcast API（rank 1~9）在 UB 内广播，无 32B 对齐限制。

DAV_2201 请使用静态接口（rank=1/2），详见 ub-broadcast.md。

一、与静态接口的对比

二、API

// 1. Kernel 侧计算 Tiling
BroadcastTiling tiling;
GetBroadcastTilingInfo<T>(rank, dstShape, srcShape, false, tiling);

// 2. 执行广播
Broadcast<T>(dstLocal, srcLocal, dstShape, srcShape, &tiling);

参数说明：

示例：

// [2, 1, 4] → [2, 3, 4]（沿 axis=1 广播，rank=3）
uint32_t dstShape[] = {2, 3, 4};
uint32_t srcShape[] = {2, 1, 4};
BroadcastTiling tiling;
GetBroadcastTilingInfo<float>(3, dstShape, srcShape, false, tiling);
Broadcast<float>(dstLocal, srcLocal, dstShape, srcShape, &tiling);

// [1, 8] → [4, 8]（沿 axis=0 广播，rank=2，无 32B 对齐要求）
uint32_t dstShape2[] = {4, 8};
uint32_t srcShape2[] = {1, 8};
BroadcastTiling tiling2;
GetBroadcastTilingInfo<half>(2, dstShape2, srcShape2, false, tiling2);
Broadcast<half>(dstLocal, srcLocal, dstShape2, srcShape2, &tiling2);

三、约束

四、数据流

与静态 UB Broadcast 相同，区别仅在 Broadcast API 调用：

GM → DataCopyPad → UB [srcShape, 未广播]
  ↓
GetBroadcastTilingInfo + Broadcast → UB [dstShape, 已广播]
  ↓
Compute → UB → DataCopyPad → GM

Tiling 参数计算、多核切分、多维索引管理与 ub-broadcast.md 完全相同。

【免费下载链接】cannbot-skills CANNBot 是面向 CANN 开发的用于提升开发效率的系列智能体，本仓库为其提供可复用的 Skills 模块。 项目地址: https://gitcode.com/cann/cannbot-skills