vllm
[Hardware] Replace torch.cuda.device_count/current_device/set_device API
#36145
Merged
Overview
Commits
10
Changes
View On GitHub

[Hardware] Replace torch.cuda.device_count/current_device/set_device API #36145

vllm-bot merged 10 commits into vllm-project:main from jikunshang:kunshang/acc_device
jikunshang
jikunshang jikunshang requested a review from tjtanaa tjtanaa 65 days ago
jikunshang jikunshang requested a review from hmellor hmellor 65 days ago
jikunshang jikunshang requested a review from ApostaC ApostaC 65 days ago
jikunshang jikunshang requested a review from orozery orozery 65 days ago
jikunshang jikunshang requested a review from jeejeelee jeejeelee 65 days ago
jikunshang jikunshang requested a review from mgoin mgoin 65 days ago
jikunshang jikunshang requested a review from tlrmchlsmth tlrmchlsmth 65 days ago
jikunshang jikunshang requested a review from WoosukKwon WoosukKwon 65 days ago
jikunshang jikunshang requested a review from yewentao256 yewentao256 65 days ago
jikunshang jikunshang requested a review from DarkLight1337 DarkLight1337 65 days ago
jikunshang jikunshang requested a review from robertgshaw2-redhat robertgshaw2-redhat 65 days ago
jikunshang jikunshang requested a review from aarnphm aarnphm 65 days ago
jikunshang jikunshang requested a review from NickLucche NickLucche 65 days ago
jikunshang jikunshang requested a review from njhill njhill 65 days ago
jikunshang jikunshang marked this pull request as draft 65 days ago
mergify
mergify mergify added documentation
mergify mergify added performance
mergify mergify added nvidia
mergify mergify added speculative-decoding
mergify mergify added v1
mergify mergify assigned sangstar sangstar 65 days ago
mergify mergify added kv-connector
gemini-code-assist
gemini-code-assist commented on 2026-03-05
mergify mergify added multi-modality
jikunshang jikunshang marked this pull request as ready for review 65 days ago
jikunshang jikunshang requested a review from ywang96 ywang96 65 days ago
jikunshang jikunshang requested a review from 22quinn 22quinn 65 days ago
jikunshang jikunshang requested a review from pavanimajety pavanimajety 65 days ago
jikunshang jikunshang requested a review from LucasWilkinson LucasWilkinson 65 days ago
jikunshang jikunshang requested a review from MatthewBonanni MatthewBonanni 65 days ago
jikunshang jikunshang added ready
jikunshang jikunshang added ready-run-all-tests
jikunshang jikunshang force pushed 62 days ago
mergify
mergify mergify added needs-rebase
jikunshang replace torch.cuda.device_count
29523ec8
jikunshang fix pre-commit
8741eea0
jikunshang replace torch.cuda.current_device
1f4f04c1
jikunshang replace torch.cuda.set_device
5365a543
jikunshang add lint for set_device
caf9c53d
jikunshang revert test_cuda_context fpr current_platform
4545ad4d
jikunshang revert.
a8dc4889
jikunshang update
2099148c
jikunshang jikunshang force pushed to 2099148c 58 days ago
mergify mergify removed needs-rebase
hmellor
hmellor requested changes on 2026-03-12
jikunshang address comments
88bc76e7
jikunshang jikunshang removed ready
jikunshang jikunshang removed ready-run-all-tests
jikunshang
hmellor
hmellor commented on 2026-03-12
hmellor
hmellor commented on 2026-03-12
hmellor
jikunshang revert
83239843
hmellor
hmellor approved these changes on 2026-03-12
hmellor hmellor enabled auto-merge (squash) 58 days ago
github-actions github-actions added ready
jikunshang jikunshang added ready-run-all-tests
jikunshang
DarkLight1337
vllm-bot vllm-bot merged 53ec16a7 into main 58 days ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
Labels
documentation performance speculative-decoding ready v1 multi-modality kv-connector nvidia ready-run-all-tests
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum