Compare commits

..

3 Commits

Author SHA1 Message Date
Walle1999 7feae82da3
Merge f6f56f0a7d into 512eed0b65 2026-06-28 18:03:49 +08:00
Walle1999 f6f56f0a7d
Merge pull request #1 from shiguol/develop/mac
feat: 添加 macOS Metal (MPS) 设备支持
2026-05-25 10:15:06 +08:00
shiguoliang 2d829fe29f feat: 添加 macOS Metal (MPS) 设备支持
通过条件分支实现 CUDA/MPS/CPU 三端兼容:
- 新增 get_default_device() 自动检测最优设备
- 混合精度、GradScaler、DDP 按设备类型条件启用
- CUDA 环境行为完全不变,MPS 作为新增分支

Co-authored-by: Cursor <cursoragent@cursor.com>
2026-05-24 10:52:56 +08:00

Diff Content Not Available