#20863 [Diffusion] Add mixed-resolution benchmark support (for #20762)
原始 PR · 作者 fengyuanyu1 · 合并时间 2026-04-22 14:22
为扩散模型 benchmark 工具添加混合分辨率请求支持,提升测试真实性。
该 PR 值得精读,特别是 `RandomDataset` 中的权重采样实现和 per-request 参数传递的设计,展示了如何扩展 benchmark 工具以支持多样化负载。关注点包括:权重采样算法(`random.choices` 的使用)、接口变更(`generate_batch` 从字典到列表)的处理、以及混合分辨率下像素吞吐量的正确计算方式。
参与讨论