| Files | Last commit | Last update |
|---|---|---|
fix broadcast_to golden Co-authored-by: t00850857<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !5391 merge broadcastto into master fix broadcast_to golden Created-by: tan_xin Commit-by: t00850857 Merged-by: cann-robot Description: ## 描述 优化 conversion/broadcast_to/tests/assets/golden.py 中 golden 函数的实现,解决 E2E 性能测试中 input shape 与 target shape 相同时无法测到真实 GPU 数据搬运的问题。 **问题原因**:torch.broadcast_to 当 input shape == target shape 时返回视图(view),不复制数据;.contiguous() 检查 tensor 已连续后直接返回自身,零拷贝。导致性能循环内 0 个 GPU kernel,测到的时间全是 Python 调度开销,无法获得真实性能基线。 **修改方法**:提取公共辅助函数 _broadcast_result,当 x.shape == target_shape(无广播维度、零拷贝场景)时额外调用 .clone() 强制触发真实数据拷贝;有广播维度时保持原实现(contiguous() 本身就会触发 copy kernel,测纯广播性能)。 **结果影响**:输出数值不变(clone 只复制数据),精度比对(binary_equal)不受影响。 ## 关联的Issue 无 ## 测试 ## 文档更新 无 ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!5391 | 1 month ago | |
[CANNBot]broadcastto算子新增单轴模板 Co-authored-by: t00850857<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !4462 merge broadcastto into master [CANNBot]broadcastto算子新增单轴模板 Created-by: tan_xin Commit-by: t00850857 Merged-by: cann-robot Description: ## 描述 本 PR 为 BroadcastTo 算子新增**单轴特化模板**:当输入/输出 tensor 经 MergeAxis 合并后维度为 1 时,跳过通用 tiling 路径,进入专用的 Tiling4SingleAxis 分核/tiling 逻辑,并由新的 BroadcastSingleAxis<T, IsBrc> 模板类在 kernel 侧执行数据传输。 通过 IsBrc 模板参数编译期区分: - **A 轴**(IsBrc=false,非广播,inDim == outDim):Double Buffer(ubPing + ubPong)交替搬入搬出,MTE2/MTE3 流水并行 - **B 轴**(IsBrc=true,标量广播,inDim != outDim):单 buffer 常驻 UB,DataCopyPad 读取标量到 scalarBuf,Duplicate(tensor版) 从 scalarBuf[0] 广播填充 UB,MTE3 循环搬出 ## 关联的Issue https://gitcode.com/cann/ops-math/issues/2755 ## 测试 ST 测试,二级冒烟 ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [x] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!4462 | 2 months ago | |
broadcastto算子tilingdata整改 Co-authored-by: tan_xin<tanxin44@huawei.com> # message auto-generated for no-merge-commit merge: !5180 merge broadcastto_tiling into master broadcastto算子tilingdata整改 Created-by: tan_xin Commit-by: tan_xin Merged-by: cann-robot Description: ## 描述 ### 命名风格整改和 workspace 清理 **原因**:命名风格不统一,workspace 参数无用。 **方法**: - brctoMaxDMADimNum → BRC_TO_MAX_DMA_DIM_NUM,brctoMaxADimNum → BRC_TO_MAX_A_DIM_NUM,brctoMaxBDimNum → BRC_TO_MAX_B_DIM_NUM,统一全大写加下划线风格 - 删除 kernel 入口 broadcast_to_impl 中的 workspace 参数和 SetSysWorkspace(workspace),同步清理 broadcast_to_apt.cpp 函数签名 - broadcast_to_tiling_arch35.cpp 同步去掉 workspace 参数 ## 关联的Issue (https://gitcode.com/cann/ops-math/issues/2755) ## 测试 - broadcast_to kernel 编译通过(ascend950) - expand/tile 编译通过 - UT 编译通过(-u --noexec) ## 文档更新 无 ## 类型标签 - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述:tiling 下沉改造 See merge request: cann/ops-math!5180 | 2 months ago | |
add template op to opensource Co-authored-by: lu-yukun<luyukun4@h-partners.com> # message auto-generated for no-merge-commit merge: !849 merge template_fanhe into master add template op to opensource Created-by: lu-yukun Commit-by: lu-yukun Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。--> <!-- 如果这个PR是为了解决特定的问题单,请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-math!849 | 9 months ago |