Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[hardware][misc] introduce platform abstraction #6080

Merged
merged 5 commits into from
Jul 3, 2024

Conversation

youkaichao
Copy link
Member

fixes #6059

the idea is to progressively absorb common if-else patterns into the vllm/platforms directory, so that we don't have scattered if-else clause for platform identification everywhere.

ideally, the platform identification is done in from vllm.platforms import current_platform , just once.

we can progressively move towards the ultimate goal, with this being the first step.

Copy link
Member

@DarkLight1337 DarkLight1337 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, let's get this in ASAP to fix the CI failure.

@youkaichao
Copy link
Member Author

merge as failed tests are not related

@youkaichao youkaichao merged commit 482045e into vllm-project:main Jul 3, 2024
68 of 70 checks passed
@youkaichao youkaichao deleted the platform branch July 3, 2024 03:13
@WoosukKwon
Copy link
Collaborator

WoosukKwon commented Jul 3, 2024

Oh sorry for the late review. I think the current_platforms is a bit misleading because it is only meaningful for AMD and NVIDIA GPUs.
Oh never mind. It's ok as we will add more devices in the future.

@youkaichao
Copy link
Member Author

yeah, the idea is to gradually add all device specific queries in this folder.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

[Bug][CI/Build]: Missing attribute 'nvmlDeviceGetHandleByIndex' in AMD tests
3 participants