lm-sys / lm-sys/FastChat

Can not use Intel Arc GPU

Open
#2,414 6 comments 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
Python
Stars
39.5k
Forks
4.8k
PR merge metrics
No merged PRs in 30d

Description

When I run `python -m fastchat.serve.model_worker --model-path lmsys/vicuna-7b-v1.5 --device xpu --gpus X`, whatever the X is, or without `--gpus`, the memory will be always loaded to both CPU and integrated GPU, and calculation will be always on CPU, single threaded.

I can confirm that GPUs can be recognized by torch:
``` python
import torch
import intel_extension_for_pytorch
torch.xpu.current_device()
# 0
torch.xpu.get_device_name(0)
# 'Intel(R) UHD Graphics 770'
torch.xpu.get_device_name(1)
# 'Intel(R) Arc(TM) A770 Graphics'
```

Any help will be great🙏

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start with the `python -m fastchat.serve.model_worker` entry point and reproduce the command using `--device xpu` with and without `--gpus`. Compare the reported torch XPU devices with where model memory and computation are placed; done means inference uses the Intel Arc GPU rather than CPU execution.

Written by the indexing model from the issue text.

Assessment

Tech stack
python, pytorch
Domain
ai
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
30/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.