修复 XPU 上 Mamba 指针高位溢出崩溃
值得精读。该 PR 是一个「小改动、大含义」的典型案例:用 3 行辅助函数解决跨平台设备指针表示差异,且通过位模式保持测试证明安全。值得关注的设计决策包括:不把 buffer 改为 uint64(保持 kernel ABI 不动,降低回归面)、复用 Python 整数范围判断规避 numpy 平台差异、以及用 `_FakeDataPtrTensor` 在无硬件环境模拟高位指针的测试技巧。后续在 XPU 上扩展其他 kernel 元数据路径时可复用该模式。
参与讨论