3
数学库
vllm 0.5.0 版本Lora推理部署报错
2024-10-16 11:38:22
166

DCU 加速卡驱动程序:rock-5.7.1-6.2.18-V1.1.2.aio.run

DCU 加速卡软件工具包:DTK-24.04.2

加速卡: K100_AI

vllm:0.5.0+das.opt1.dtk24042

vllm启动参数:
python -m vllm.entrypoints.openai.api_server
--port 9009
--model /models
--lora-modules vllm-lora=/lora
--tensor-parallel-size 1
--served-model-name vllm-model --trust-remote-code --max-model-len 4096 --enable-lora --max-lora-rank 8

报错内容如下:

[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/model_executor/models/qwen2.py", line 228, in forward
[rank0]: hidden_states = self.mlp(hidden_states)
[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in _call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/model_executor/models/qwen2.py", line 80, in forward
[rank0]: gate_up, _ = self.gate_up_proj(x)
[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1532, in _wrapped_call_impl
[rank0]: return self._call_impl(*args, **kwargs)
[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/nn/modules/module.py", line 1541, in call_impl
[rank0]: return forward_call(*args, **kwargs)
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/lora/layers.py", line 470, in forward
[rank0]: output_parallel = self.apply(input
, bias)
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/lora/layers.py", line 600, in apply
[rank0]: _apply_lora_packed_nslice(
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/lora/layers.py", line 129, in _apply_lora_packed_nslice
[rank0]: add_lora_slice(output, x, lora_a_stacked[slice_idx],
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/lora/punica.py", line 196, in add_lora_slice
[rank0]: ops.dispatch_bgmv_low_level(
[rank0]: File "/usr/local/lib/python3.10/site-packages/vllm/_custom_ops.py", line 580, in dispatch_bgmv_low_level
[rank0]: torch.ops._punica_C.dispatch_bgmv_low_level(
[rank0]: File "/usr/local/lib/python3.10/site-packages/torch/ops.py", line 854, in call
[rank0]: return self
._op(*args, **(kwargs or {}))
[rank0]: RuntimeError: No suitable kernel. h_in=8 h_out=18944 dtype=Float out_dtype=BFloat16

全部评论(4)
只看楼主
网
网友po20TW
楼主
1楼
模型是qwen2-7b
2024/10/17
Z
zhuwenwen
回复 @网友po20TW
这个在0.5.0上有点问题,需要0.6版本支持
2024/10/18
L
luck
回复 @zhuwenwen
dcu 对应的0.6在哪里可以下载
3/31
M
mjglmvuid0dt
回复 @luck
可以参考:https://das.sourcefind.cn:55011/portal/#/installation?id=9385225c-6b7a-11ef-bf92-005056904552&type=frame和https://developer.sourcefind.cn/servicelist/detail?post_id=61036870-b3c7-11f0-9989-acde48001122&active=TagDownload
4/1
没有更多内容了
--
楼主
Level.2
0 帖子
楼主热帖
暂无数据