✨[Feature] Device Auto-Detection in Torch Compile Path #2172

Closed

Closed

✨[Feature] Device Auto-Detection in Torch Compile Path#2172

Assignees

Labels

feature request

opened

Add support for Device specification in torch.compile path, avoiding the need for CUDA_VISIBLE_DEVICES or similar
Add support for auto-detection of device based on model parameters

Metadata

Assignees

gs-olive

Labels

feature request

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests