SuperAPI AI 工具集

wheel-size-analyzer

测试与质量

查看项目

更新于 10-16 04:54

技能介绍

用途:追踪并报告 PyTorch nightly wheel 体积随时间的变动,帮助定位体积回归发生的时间点、比较不同 nightly 构建的体积,并为 manywheel 各构建变体生成体积报告。

主要能力:按日期范围查询 nightly 工作流运行与制品大小;按 CUDA 版本/变体(cpu、xpu、cuda12_8、cuda13_0、rocm6_4 等)映射到制品名 manywheel-py3_12-{variant};统计起止体积、总变化与百分比;列出日环比超过阈值(默认 1 MB)的显著跳变;生成每日明细与关键观察。

使用方式:需确认三个参数——日期范围(必填)、CUDA 构建或变体(默认 cuda12_8)、体积阈值(默认 1 MB)。查询使用 gh api 调用 pytorch/pytorch 工作流 21257348(linux-binary-manywheel),例如:gh api "repos/pytorch/pytorch/actions/workflows/21257348/runs?branch=nightly&created=YYYY-MM-DD&per_page=1",再以 gh api "repos/pytorch/pytorch/actions/runs/{run_id}/artifacts?per_page=100" 获取 size_in_bytes 并换算为 MB。可跨月并行收集数据后合并报告;调查具体跳变时可比较 nightly 提交的 base SHA 与提交列表,并用构建日志或跨变体对比验证成因。

适用场景:跟踪二进制体积变化、识别体积回归、比较 nightly 构建、为 manywheel 变体产出体积报告,以及排查某日体积跳变的原因。

限制:依赖 GitHub Actions 制品 API 与 nightly 工作流数据,部分日期可能无运行或制品缺失(构建失败或制品临时改名),需在报告中标注;许多 manywheel 构建并不针对主线每次提交运行,逐提交验证可能不可用;制品大小为预压缩的 zip 大小;节假日期间可能缺少运行记录。资料未提供其他收费、价格或平台信息。