VLLM适配昇腾开源验证任务
1 背景介绍
开源for Huawei(参考:https://developer.huaweicloud.cn/programs/opensource/contributing/) 通过和公司、高校、社区的开发者合作,完成鲲鹏、昇腾、欧拉、鸿蒙、高斯、云服务等与开源软件的适配开发,帮助繁荣Huawei的基础生态,同时让开源软件能够更加简单、高效的运行于华为云上。
开始之前,开发者可以下载 开源for Huawei Wiki(参考:https://gitcode.com/HuaweiCloudDeveloper/OpenSourceForHuaweiWiki/overview) 了解详细的开发步骤,技术准备,以及开发过程需要的各种资源。
2 需求分析
2.1 开源项目基本情况
VLLM是一个开源的高性能大语言模型推理库,旨在实现大规模语言模型的快速推理。截至到2024年11月25日,VLLM项目的GitHub star 数量超过 30.7k,项目代码最新提交时间为2024年11月24日。
源码地址:https://github.com/vllm-project/vllm
官网主页:https://github.com/vllm-project
主要开发语言:python
LICENSE:Apache License 2.0
维护者:Woosuk Kwon
项目数据:Fork 4.7k,Star 30.7K,Contributor 706,最近一次提交:2024-11-24
star历史(https://star-history.com/):
2.2 任务目的和范围
本任务旨在将 VLLM 项目适配至华为的 Ascend和 Kunpeng处理器以及Euler操作系统,确保其能够在这些平台上有效运行,有助于推动AI计算基础设施的发展。
- 能力要求
完成该任务需要掌握python语言,具备昇腾模型迁移适配能力,掌握昇腾硬件上开发环境的搭建流程、模型迁移方法(自动迁移、工具迁移)等。
- 适配任务清单
生态 |
生态细类 |
是否需要验证 |
验证逻辑说明 |
是否需要适配 |
适配场景说明 |
昇腾生态 |
指令 |
是 |
使用昇腾处理器运行应用。 |
否 |
使用昇腾指令实现适配逻辑。 |
CANN |
是 |
应用功能依赖于CANN运行。 |
否 |
使用CANN实现适配逻辑。 |
|
MindSpore
|
否 |
应用功能依赖于MindSpore运行。 |
否 |
使用MindSpore实现适配逻辑 |
|
torch_npu |
是 |
应用功能依赖于torch_npu运行 |
否 |
使用torch_npu实现适配逻辑 |
|
鲲鹏生态 |
指令 |
是 |
使用鲲鹏处理器运行应用。 |
否 |
使用鲲鹏指令实现适配逻辑。 |
Euler |
是 |
使用Euler系统运行应用。 |
否 |
使用Euler系统调用实现适配逻辑。 |
|
存储生态 |
OBS |
否 |
使用OBS功能。 |
否 |
基于OBS API访问实现适配逻辑。 |
鸿蒙生态 |
HarmonyOS |
否 |
使用HarmonyOS系统运行应用。 |
否 |
使用HarmonyOS系统调用实现适配逻辑。 |
3 工作量评估
任务 |
人/天 |
在昇腾、鲲鹏处理器和欧拉操作系统上的环境配置与测试 |
4 |
完成Demo设计与功能验证 |
2 |
撰写任务和心得博客并发布 |
2 |
总计 |
8人天 |
4 任务分解
4.1 完成鲲鹏处理器、Euler系统、昇腾NPU适配功能开发
在GitCode完成项目Fork,鲲鹏处理器、Euler系统、昇腾NPU适配需要的脚本和代码开发,以确保项目能够充分利用这些硬件平台的优势,实现高效稳定的运行。
4.2 完成DEMO设计
流程图参考如下:
4.3 完成 VLLM演示 Demo 的设计、开发和部署
- 模型选择:用户通过github/huggingface获取VLLM需要的模型。
- 模型推理:用户根据场景输入符合要求的文本数据,VLLM模型根据输入的数据预测一个最终的结果展现给用户。
- 适配与优化:确保VLLM模型在Ascend、Kunpeng处理器及 openEuler 系统上稳定高效运行处理器上稳定高效运行。
- 提供详细的部署文档,列出步骤,便于用户在Ascend、Kunpeng和openEuler环境中复现Demo。
4.4 完成适配心得输出
撰写一篇详细的适配心得,内容包括 VLLM项目在 Ascend 和 Kunpeng 处理器上的适配过程、面临的技术挑战、解决方案。心得文档将发布于华为云开发者论坛博客。
内容包括不限于如下内容:
l 开源for Huawei的背景
l 开源软件的功能介绍
l 开发适配过程及碰到和解决的问题
l DEMO设计、开发、部署过程
博客打上如下标签:
l 开源(必选)
l 开发者(必选)
l 鲲鹏(可选)
l 昇腾(可选)
l Euler(可选)
5 验收材料
验收细则 |
评分说明 |
是否为验收必选项 |
结果(通过/不通过) |
开发者举证 |
功能开发:完成与华为三大根技术生态(鲲鹏云、昇腾云、鸿蒙)适配的功能开发 |
基于鲲鹏、昇腾和Euler系统完成验证。 |
是 |
|
|
代码质量:适配华为三大根技术生态的代码质量经过工具检测达标 |
推送给开源社区的代码符合开源社区关于代码风格、功能、自动化测试等要求。 |
否 |
|
|
提供相关文档,比如部署文档、使用文档、配置文档等 |
根据社区要求,完善相关的开发指南和部署指导 |
否 |
|
|
合入到开源项目主仓:代码推送上游社区 |
代码合并到当前主干分支。举证提供代码仓库链接。 |
否 |
|
|
Deomo设计&开发:完成功能演示Demo的开发 |
举证提供DEMO仓库地址。 |
是 |
|
|
Demo部署&验证:完成功能演示DEMO基于华为昇腾、鲲鹏和Euler系统环境的部署和功能验证 |
提供DEMO部署验证结果的关键截图进行举证。提供DEMO运行环境和资源的规格清单。 |
是 |
|
|
发布博客:发布博客介绍适配心得。 |
提供博客的地址。 |
是 |
|
|
制作单机版无高危开源镜像。 |
上架云商店。 |
否 |
|
|
单机版镜像上架时支持模板部署。 |
|
否 |
|
|
制作集群版无高危开源镜像 |
上架云商店。 |
否 |
|
|
集群版镜像上架时支持模板部署 |
|
否 |
|
|
6 参考资料
1.VLLM Github仓库: https://github.com/vllm-project/vllm
2.华为 Ascend 开发者社区: https://developer.huawei.cn/ascend
3.华为 Kunpeng 开发者社区: https://developer.huawei.cn/kunpeng
4.openEuler 官方网站: https://www.openeuler.org
- 点赞
- 收藏
- 关注作者
评论(0)