Prhub

#46095 [Model Runner V2] Fix MRv2 memory leak test

原始 PR 作者 yewentao256 合并时间 2026-06-19 08:36 文件变更 1 提交数 1 评论 0 代码增减 +1 / -1

执行摘要

修复 MRv2 内存泄漏测试,调大 warmup 轮次

MRv2 在第二轮 warmup 时会额外分配约 2MB 显存,之后保持稳定,测试误判为内存泄漏。PR 作者在 body 中说明:"MRv2 will allocate additional 2MB in second round warm up, then keep stable, which doesn't have memory leak at all."

修改合理且安全,无需深入分析。可快速合并。

讨论亮点

无 review 讨论。

实现拆解

仅修改测试配置:将 tests/models/multimodal/generation/test_memory_leak.py 中的 WARMUP_ROUNDS 从 1 改为 2。其他所有逻辑、断言和阈值均保持不变。

文件 模块 状态 重要度
tests/models/multimodal/generation/test_memory_leak.py 测试 modified 3.23

分析完成后,这里会展示 LLM 生成的相对完整源码片段和详细注释。

评论区精华

没有提炼出高价值讨论线程

当前评论区没有形成足够清晰的争议点或结论,后续有更多讨论时会体现在这里。

风险与影响

极低风险。仅调整 warmup 轮次,不影响任何生产代码或测试逻辑的正确性。若未来 MRv2 出现真实内存泄漏,此修改可能使首次 warmup 后的泄漏被掩盖(需第二轮 warmup 后才检测),但测试已包含多轮测量,实际泄漏仍会被后续 round 的增长检出。

直接影响 MRv2 环境下的内存泄漏测试:消除误报,但可能略微降低对首轮后缓慢泄漏的敏感度。对其他模型或 V1 runner 无影响。

仅测试配置调整

关联 Issue

未识别关联 Issue

当前没有检测到明确关联的 Issue 链接,后续同步到相关引用后会出现在这里。

完整报告

参与讨论