Skip to content

[ROCm] fix get_device_name for rocm#13438

Merged
DarkLight1337 merged 2 commits intovllm-project:mainfrom
divakar-amd:fix_device_name_amdsmi
Feb 18, 2025
Merged

[ROCm] fix get_device_name for rocm#13438
DarkLight1337 merged 2 commits intovllm-project:mainfrom
divakar-amd:fix_device_name_amdsmi

Conversation

@divakar-amd
Copy link
Contributor

@divakar-amd divakar-amd commented Feb 18, 2025

Use amd-smi to get the device name for AMD GPUs.

Tested using:

VLLM_ENABLE_V1_MULTIPROCESSING=0 VLLM_USE_V1=1 python benchmark_latency.py --model Mixtral-8x7B-v0.1/ --load-format dummy -tp 8

@github-actions
Copy link

👋 Hi! Thank you for contributing to the vLLM project.

💬 Join our developer Slack at https://slack.vllm.ai to discuss your PR in #pr-reviews, coordinate on features in #feat- channels, or join special interest groups in #sig- channels.

Just a reminder: PRs would not trigger full CI run by default. Instead, it would only run fastcheck CI which starts running only a small and essential subset of CI tests to quickly catch errors. You can run other CI tests on top of those by going to your fastcheck build on Buildkite UI (linked in the PR checks section) and unblock them. If you do not have permission to unblock, ping simon-mo or khluu to add you in our Buildkite org.

Once the PR is approved and ready to go, your PR reviewer(s) can run CI to test the changes comprehensively before merging.

To run CI, PR reviewers can either: Add ready label to the PR or enable auto-merge.

🚀

Signed-off-by: Divakar Verma <divakar.verma@amd.com>
@divakar-amd divakar-amd force-pushed the fix_device_name_amdsmi branch from 76fb663 to 67308e4 Compare February 18, 2025 00:16
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
@DarkLight1337 DarkLight1337 enabled auto-merge (squash) February 18, 2025 02:34
@github-actions github-actions bot added the ready ONLY add when PR is ready to merge/full CI is needed label Feb 18, 2025
@DarkLight1337 DarkLight1337 merged commit 7c7adf8 into vllm-project:main Feb 18, 2025
57 checks passed
xjpang pushed a commit to xjpang/vllm that referenced this pull request Feb 20, 2025
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Akshat-Tripathi pushed a commit to krai/vllm that referenced this pull request Mar 3, 2025
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
lulmer pushed a commit to lulmer/vllm that referenced this pull request Apr 7, 2025
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Signed-off-by: Louis Ulmer <ulmerlouis@gmail.com>
shreyankg pushed a commit to shreyankg/vllm that referenced this pull request May 3, 2025
Signed-off-by: Divakar Verma <divakar.verma@amd.com>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ready ONLY add when PR is ready to merge/full CI is needed

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants