#32984 [MLX] Upgrade to Torch 2.13/MLX 0.32+ and redesign the Torch-MLX tensor bridge
原始 PR · 作者 yeahdongcn · 合并时间 2026-08-22 09:51
升级 Torch 2.13/MLX 0.32,重设计零拷贝张量桥
值得精读,尤其是 `python/sglang/srt/utils/tensor_bridge.py` 的所有权契约(owned vs borrowed 三个层次)与负 stride 物化设计:它把「双框架共用同一块 Metal 内存」的竞态、生命周期问题显式化,并用 RLock 串行化完整穿越、用 DLPack 消除 payload 拷贝。对任何做 Torch + 其他运行时张量桥的团队都有直接借鉴意义;`runtime.py` 的版本门控矩阵(拒 rc/dev、Torch 锁系列、MLX 只设下限)也是可复用的 fail-fast 模式。
参与讨论