mirror of
https://github.com/kvcache-ai/ktransformers.git
synced 2026-11-05 01:07:28 +00:00
[GH-ISSUE #984] [Bug] 不支持海光DCU K100A (gfx928) #698
Labels
No milestone
No project
No assignees
1 participant
Notifications
Due date
No due date set.
Dependencies
No dependencies set.
Reference
kvcache-ai/ktransformers#698
Loading…
Add table
Add a link
Reference in a new issue
No description provided.
Delete branch "%!s()"
Deleting a branch is permanent. Although the deleted branch may continue to exist for a short time before it actually gets removed, it CANNOT be undone in most cases. Continue?
Originally created by @yeungtuzi on GitHub (Mar 27, 2025).
Original GitHub issue: https://github.com/kvcache-ai/ktransformers/issues/984
检查清单
问题描述
在make dev_install的时候报错
似乎是编译器坚持要编译gfx908等amdgpu架构代码,但是忽略了本地实际上是gfx928
但是软件检测又正常
复现步骤
按照rocm.md操作,正常编译到54/54之后的1/3就会出错
环境信息
硬件 环境
AMD Ryzen Threadripper PRO 7975WX 32-Cores
512G RAM
软件 环境
ubuntu 22.04
anaconda create -n kt python=3.11
conda activate kt
cuda-toolkit-12.8
海光按照官方文档安装驱动
(kt) root@lf:~/ktransformers# hycc -v
@yeungtuzi commented on GitHub (Mar 27, 2025):
检索代码,发现没有跟gfx928相关的内容
(kt) root@lf:~/ktransformers# grep "gfx9" -R *
但是我在海光提供的DAS目录中找到了相关定义
(kt) root@lf:/opt# ls -ld rocm
lrwxrwxrwx 1 root root 18 Mar 27 04:40 rocm -> dtk-25.04-DCC-2503
grep gfx928 -R * rocm/
@south-ocean commented on GitHub (Mar 27, 2025):
你好,ktransformers 海光代码分支是基于DTK和DAS软件栈编译的,需要安装DAS软件栈配套的AI whl包,尤其是DAS下面的torch和fastpt,然后可以实现ktransformer cuda代码的快速兼容适配。
你可以先确认下环境是否配置好,比如torch、apex、fastpt等是否安装完成并可以正常导入,并且激活dtk下面的cuda env.sh,实现cuda代码兼容。具体编译方式和软件栈资源可见开发者社区https://developer.sourcefind.cn/。
目前可有开箱即用的容器方式使用,见https://forum.sourcefind.cn/thread/980。
@yeungtuzi commented on GitHub (Mar 28, 2025):
编译已经成功,目前试验性跑一下,似乎flashinfer没有rocm/hygon dtk的支持?那么是否意味着后续多并发就没辙了
@yeungtuzi commented on GitHub (Mar 28, 2025):
已经完成编译并可以运行,有胡言乱语的问题,提交了新的issue,本issue似乎可以关闭了