#51440 [CI Test] Add specific unit test for mrv2 offloading
原始 PR · 作者 yewentao256 · 合并时间 2026-08-08 06:49
为 MRv2 offload 新增定向单测并回退 CI 变更
值得快速阅读,重点看 tests/basic_correctness/test_cpu_offload.py 中的 test_mrv2_weight_offloading:它展示了如何穿透 engine_core 到 GPUModelRunner,并通过 get_offloader() 直接校验 offloader 内部状态,这种测试思路对验证 MRv2 内部机制很有参考价值。对于负责 offload 或 MRv2 功能的工程师,建议精读并留意该测试对内部实现的耦合度。
参与讨论