Skip to content

feat: Apple Silicon MPS support (bf16, no autocast) - #56

Open
yhsung wants to merge 1 commit into
OpenMOSS:mainfrom
yhsung:main
Open

yhsung wants to merge 1 commit into
OpenMOSS:mainfrom
yhsung:main

Conversation

@yhsung

@yhsung yhsung commented Sep 7, 2026

Copy link
Copy Markdown
  • resolve_device auto now prefers cuda, then mps, then cpu
  • pass device through to audio processor on mps
  • keep autocast cuda-only: autocast on mps produces degenerate all-zero output, bf16 weights run correctly without it
  • README dtype example updated for mps

Measured on M-series Mac: 10-min audio in ~4:20 (RTF ~0.44), ~2.4x faster than CPU, identical segments.

Closes fork ahead by 1 commit (dc42ba3).

- resolve_device auto now prefers cuda, then mps, then cpu
- pass device through to audio processor on mps
- keep autocast cuda-only: autocast on mps produces degenerate
  all-zero output, bf16 weights run correctly without it

Measured on M-series Mac: 10-min audio in ~4:20 (RTF ~0.44),
~2.4x faster than CPU, identical segments.
@yiakwy-xpu-ml-framework-team

yiakwy-xpu-ml-framework-team commented Sep 15, 2026

Copy link
Copy Markdown

I am adding support of MOSS to velox voice https://github.com/yiakwy-xpu-ml-framework-team/velox-voice, then you can use it in latest M5 Ultra, stay tuned. @yhsung

Note MPS is not good choice for apple applicaiton, since MPS does not utilize ANE and unified memory features in apple efficienly.

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