cnpc-chem-opt:基于 UniCore 的 softmax_dropout 模块跨后端重构项目

用户可通过该项目获取 UniCore softmax_dropout 模块的跨后端重构方案,实现 CPU、CUDA、Ascend NPU 等多后端支持,解决非 CUDA 后端兼容性问题,提供零回归补丁及完整测试验证。【此简介由AI生成】

分支2Tags0

项目介绍

用户可通过该项目获取 UniCore softmax_dropout 模块的跨后端重构方案,实现 CPU、CUDA、Ascend NPU 等多后端支持,解决非 CUDA 后端兼容性问题,提供零回归补丁及完整测试验证。【此简介由AI生成】

定制我的领域