0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robot泛化工程：单核切K系列模板
ca9da0df创建于 2025年12月19日历史提交
文件最后提交记录最后更新时间
do_tiling_b16.h
【task】更新license Co-authored-by: yuantao<taoyuan15@h-partners.com> # message auto-generated for no-merge-commit merge: !399 merge master into master 【task】更新license Created-by: yuantao_ Commit-by: yuantao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3995 个月前
dynamic_optimized_matmul.h
【task】更新license Co-authored-by: yuantao<taoyuan15@h-partners.com> # message auto-generated for no-merge-commit merge: !399 merge master into master 【task】更新license Created-by: yuantao_ Commit-by: yuantao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3995 个月前
platform_info.h
【task】更新license Co-authored-by: yuantao<taoyuan15@h-partners.com> # message auto-generated for no-merge-commit merge: !399 merge master into master 【task】更新license Created-by: yuantao_ Commit-by: yuantao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3995 个月前
select_kernel_b16.h
【task】更新license Co-authored-by: yuantao<taoyuan15@h-partners.com> # message auto-generated for no-merge-commit merge: !399 merge master into master 【task】更新license Created-by: yuantao_ Commit-by: yuantao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3995 个月前
tiling_params.h
泛化工程：单核切K系列模板 Co-authored-by: Yao Weijie<yaoweijie2@h-partners.com> # message auto-generated for no-merge-commit merge: !358 merge singlecore-splitk into master 泛化工程：单核切K系列模板 Created-by: lavateinn Commit-by: Yao Weijie Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 泛化工程：新增单核切K系列模板 Block层 1. 每次传入的粒度为A(m1, k1), B(k1, n1), C(m1, n1)，然后再block层中在L1上对ABC矩阵按照L0的粒度进行切分，实际分成CeilDiv(m1, m0) * CeilDiv(n1, n0)次计算 2. L1上A和B的STAGES独立配置，复用L1A时，L1A不开db， L1A_STAGES=1，L1B_STAGES=2，复用L1B时同理 3. L0C支持unitFlag和doubel buffer写出，当L0C_STAGES=1时 可配置unitFlag=true/false,当L0C_STAGES>1时，unitFlag必须为false，L0C是否开db影响 L0的tileshape 4. 传入isAtomicAdd参数控制写出时是否在gmC上原子加 5. 在泛化工程中传入动态的tiling需要 1. SingleCoreSplitkForSmallK和PaddingSingleCoreSplitkForSmallK （非padding和padding版本） 针对 K <= k1 的特殊场景，对每个C block写出一次 ElementC；padding版本gmWC分配方式为全局，计算时写出到gmWC，计算完成后vector从gmWC搬运到gmC 2. PaddingSingleCoreSplitkKLoopOuter K循环外置的版本 K > k1 的场景，写出类型为ElementAccumulator类型，在gmC上原子加，分核策略是按照C矩阵的block平铺分核每个核处理 CeilDiv(M, m1) * CeilDiv(N, n1) / coreNum个C基本块，gmWC 分配方式为全局 3. PaddingSingleCoreSplitkKLoopMiddle K循环中置的版本 K > k1的场景，写出类型为ElementAccumulator类型，在C矩阵上按照 (m1*m1Factor, n1*n1Factor)的大块粒度进行swizzle和分核，每次按照单核切K的方式完成 (m1*m1Factor, n1*n1Factor)粒度的局部结果，再处理下一个大块； 在计算大块的时候，m1Factor <= n1Factor的时候复用A，m1Factor > n1Factor复用B； gmWC分配方式为局部，分配了m1*m1Factor*n1*n1Factor*2*coreNum的内存，每次处理大块的时候写出到大块大小的gmWC上，计算完成一个大块时vector将这一块的结果搬运到gmC上 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3585 个月前
utils.h
【task】更新license Co-authored-by: yuantao<taoyuan15@h-partners.com> # message auto-generated for no-merge-commit merge: !399 merge master into master 【task】更新license Created-by: yuantao_ Commit-by: yuantao Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #123--> ## 原因 <!--说明此次改动的目的、解决的问题等，应与类型标签匹配 --> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/catlass!3995 个月前