| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
init blaslt Co-authored-by: wangzitao_leo<wangzitao4@huawei.com> # message auto-generated for no-merge-commit merge: !3 merge init_blaslt into master init blaslt Created-by: wangzitao_leo Commit-by: ZitaoWang;wangzitao_leo Merged-by: QK_25415 Description: init blaslt See merge request: cann/ops-blas!3 | 4 个月前 | |
Fix:Blaze-ASC compat bug Co-authored-by: Zitao Wang<wangzitao4@huawei.com> # message auto-generated for no-merge-commit merge: !215 merge blaze_compat_debug into master Fix:Blaze-ASC compat bug Created-by: wangzitao_leo Commit-by: Zitao Wang Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动,包括改动的原因和所采取的方法。--> 本次 PR 针对 Blaze 与 ASC(昇腾)硬件平台之间的类型兼容性问题进行修复,关联 Issue #221。核心工作是移除此前为弥合 ops-tensor(Blaze)头文件与 CANN 9.1.0 SDK 之间 FP4/FP8 类型定义冲突而引入的自定义补丁机制,包括删除 533 行的 block_mmad_mx.h 补丁头文件及对应的 CMake 补丁函数,并调整 MXFP4/MXFP8 矩阵乘法内核代码以直接使用 Blaze 原生模板接口。 主要改动 1. 删除自定义 Blaze 补丁头文件:移除 cmake/patches/ops-tensor/block_mmad_mx.h(533 行),该文件曾为 BlockMmad 等 Blaze 模板提供 ASC 平台兼容性覆盖定义,现已不再需要。 2. 移除 CMake 补丁逻辑:删除 cmake/patches/ops-tensor/patch.cmake 中的 ops_blas_patch_ops_tensor_fp4_types 函数(59 行),该函数曾用于修改 ops-tensor 的 macro_impl.h 和 kernel_qbmm_mx.h 以解决 FP4 类型重定义冲突;同步简化 cmake/third_party/ops-tensor.cmake 中对补丁函数的调用。 3. MXFP4 内核适配调整:matmul_mxfp4_kernel.cpp 中重构模板实例化方式,将原先通过宏 MXFP4_DEFINE_TRANS_SET 批量生成的独立 kernel 函数改为由统一的 RunQuantMatmulMxfp4Kernel 模板函数驱动,直接引用 Blaze 的 BlockMmad、GemmUniversal 等原生模板,移除对补丁头文件的依赖。 4. MXFP8 内核适配调整:matmul_mxfp8_kernel.cpp 进行与 MXFP4 类似的适配,将 QuantMatmulMxfp8Kernel 改为直接使用 Blaze 原生 BlockMmad<DispatchPolicy, AType, LayoutA, ...> 等模板组合,并在 matmul_mxfp8_kernel_do 中以内联条件分支替代原先的宏展开方式进行 kernel 启动分发。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue,请在这里提供Issue链接。例如:关联Issue #000--> 关联Issue [#221](https://gitcode.com/cann/ops-blas/issues/221) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> bash build.sh --ops=blasLtMatmul --soc=ascend950  ## 文档更新 <!--如果这个PR包含文档的更新,请在这里指出。例如:更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他,请描述: See merge request: cann/ops-blas!215 | 16 天前 |
| 文件 | 最后提交记录 | 最后更新时间 |
|---|---|---|
| 4 个月前 | ||
| 16 天前 |