Skip to content

Add MPS device selection to demos - #172

Open
mariano wants to merge 1 commit into
bytedance:masterfrom
mariano:codex/add-mps-demo-device-selection
Open

Add MPS device selection to demos#172
mariano wants to merge 1 commit into
bytedance:masterfrom
mariano:codex/add-mps-demo-device-selection

Conversation

@mariano

@mariano mariano commented May 16, 2026

Copy link
Copy Markdown

On metal devices current dolphin demo forces CPU binding. This PR adds a shared demo device helper that selects CUDA, then Apple MPS, then CPU.

@mariano
mariano marked this pull request as ready for review May 16, 2026 13:24
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.

1 participant