Custom_ops:基于 Ascend C 的自定义融合算子项目

可在华为 Ascend 910B 上实现高效算子融合,提升 AI 模型性能。项目提供 FusedBiasGelu 和 FusedAddMul 算子,通过 Ascend C 编写 kernel,减少 HBM 访问开销,延迟较未融合算子显著降低。【此简介由AI生成】

分支1Tags0

项目介绍

可在华为 Ascend 910B 上实现高效算子融合,提升 AI 模型性能。项目提供 FusedBiasGelu 和 FusedAddMul 算子,通过 Ascend C 编写 kernel,减少 HBM 访问开销,延迟较未融合算子显著降低。【此简介由AI生成】

定制我的领域