feat(musa): 添加对MUSA设备的支持
添加musa.txt依赖文件并修改setup.py以支持MUSA设备 包括添加_is_musa检查函数,更新版本信息和构建选项
This commit is contained in:
9
requirements/musa.txt
Normal file
9
requirements/musa.txt
Normal file
@@ -0,0 +1,9 @@
|
||||
# Common dependencies
|
||||
-r common.txt
|
||||
|
||||
numba == 0.61.2 # Required for N-gram speculative decoding
|
||||
|
||||
# Dependencies for NVIDIA GPUs
|
||||
ray>=2.48.0 # Ray Compiled Graph, required for pipeline parallelism in V1.
|
||||
torch==2.7.1
|
||||
triton == 3.1.0 # FIXME(woosuk): This is a hack to avoid import error.
|
||||
Reference in New Issue
Block a user