加速替代CUDA!摩爾線程開源vLLM-MUSA 加速國產GPU AI
2024-11-05 20:37來源:快科技編輯:時寒峰
原標題:加速替代CUDA!摩爾線程開源vLLM-MUSA 加速國產GPU AI
11月5日消息,近日,摩爾線程上線了大語言模型高速推理框架開源項目vLLM的MUSA移植版本,為開發者提供基于摩爾線程全功能GPU進行開源項目MUSA移植的范例。
摩爾線程表示,正努力圍繞自主研發的統一系統架構GPU、MUSA軟件平臺,構建完善好用的MUSA應用生態。
vLLM是一個高效且易用的大模型推理和服務框架,支持創新的PagedAttention內存管理、連續批處理請求、CUDA/HIP圖的快速模型執行、CUDA核心優化、分布式推理支持等,可顯著提高大語言模型(LLM)的推理性能。
目前,vLLM已在各種大語言模型中得到廣泛應用,成為業界常用的開源大模型推理框架。
摩爾線程基于vLLM v0.4.2版本,進行了移植適配,使之支持摩爾線程GPU后端Device,并完全開源,開發者可以二次開發、升級vLLM到社區更新版本等。

值得一提的是,得益于摩爾線程MUSA架構的先進性、軟件棧對CUDA的優良兼容性,通過MUSIFY代碼自動轉換工具,用戶可以快捷地將原有的CUDA代碼遷移到MUSA平臺,實現替代。
同時,CUDA相關庫的調用,也可以快速替換為相應的MUSA加速庫,如算子庫muDNN、集合通信庫MCCL、線性代數庫muBLAS等。
摩爾線程通過MUSA軟件棧對CUDA軟件棧接口兼容,大幅提升了應用移植的效率,縮短了開發周期,還提供MUSIFY自動代碼移植工具等一系列實用工具和腳本。
?????投稿郵箱:jiujiukejiwang@163.com ??詳情訪問99科技網:http://www.hacbq.cn
相關推薦
推薦資訊























