Skip to content

Conversation

vshekhawat-hlab
Copy link
Contributor

@vshekhawat-hlab vshekhawat-hlab commented Mar 14, 2025

Motivation

Currently, the device argument in ServerArgs defaults to "cuda", which forces the use of CUDA even when other hardware accelerators (like XPU, HPU, or CPU) may be available or preferred. This PR aims to make the device selection more dynamic and flexible by automatically detecting the available device when no explicit device is specified.

Modifications

Changed device default from "cuda" to None
If device is not specified, get_device() will automatically detect and set the appropriate device (cuda, xpu, hpu).

Checklist

@merrymercy merrymercy merged commit bf63ee5 into sgl-project:main Mar 16, 2025
19 of 21 checks passed
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.

2 participants