Skip to content

Main dev for search#18

Open
laksjdf wants to merge 45 commits into
jiashaokun-1:mainfrom
laksjdf:main_dev_for_search
Open

Main dev for search#18
laksjdf wants to merge 45 commits into
jiashaokun-1:mainfrom
laksjdf:main_dev_for_search

Conversation

@laksjdf

@laksjdf laksjdf commented Jun 1, 2026

Copy link
Copy Markdown
Contributor

寻优优化,支持指定过滤条件和排序依据

vuemen and others added 30 commits May 26, 2026 16:35
- 在TrainingConfig中新增tp_coc和tp_coc_tile_k配置选项
- 在comm_inserter中为TP通信节点添加CoC注解和属性
- 修改compute_exposed_comm_time以支持CoC重叠模式计算
- 更新ChromeTraceExporter以在可视化中偏移CoC通信开始时间
- 在modeller和CLI中暴露tp_coc参数
前端采集用户名并展示任务提交统计
launcher 改为浅色仪表盘风格,与报告统一
每个 Pareto 条目带"open/render"按钮,Top-1 在 search 完成时同步渲染
HTML,其他条目首次点击触发 POST /search/{job_id}/render/{idx} 按需
生成,URL 缓存在 job 结果中。共享 estimate 的 export_estimate_html。

- _grid_search_with_strategies / _pareto_frontier_with_strategies:
  保留 (strategy, report) 对,原版 grid_search 丢了 strategy 无法重渲
- _render_strategy_detail:build_graph + op_cost + estimate + export
- _search_states 独立 dict 存 Strategy(不可 JSON 序列化)
- _resolve_job_artifact 同时认 pareto_frontier[*].html_filename
- 前端:详情列按钮 + 内联展开 All N 视图(替代原 iframe 报告)
- SearchRequest.output_dir 新字段
【feature】TP掩盖:为TP all_reduce添加通信计算重叠(CoC)支持
在 PPStitchedTimeline 中新增 phase_breakdown 方法,通过时间线任务精确计算每个阶段(warmup、steady、cooldown)的前向和后向时间。更新 TrainingPipelinePass 以使用这些精确值替代基于瓶颈的近似值,并在 PipelineStepMetrics 中新增相应字段以支持更详细的性能分析。
zty123456 and others added 15 commits June 1, 2026 14:46
【feature】添加基于时间线的训练阶段细粒度指标
更新训练寻优报表表头
- 将 `pipeline_metrics` 替换为统一的 `step_result` 字典,由 `StepResult.to_dict_ms()` 生成
- 移除 `PipelineStepMetrics` 类及相关导入,简化代码结构
- 删除已弃用的 `--pp-mode` CLI 参数,仅保留基于网格的 `PPStitcher` 模式
- 重命名 `recompute_time_ms` 和 `recompute_time_raw_ms` 为 `recompute_critical_ms` 和 `recompute_raw_mag_ms`,以更清晰地反映其含义
- 更新所有测试、导出器和报告生成代码以使用新的字段名和数据结构
- 在 `TrainingPipelinePass` 中直接计算重计算指标,确保与流水线调度结果一致
寻优优化,支持指定过滤条件和排序依据
【refactor】统一生成trace并移除旧模式
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

6 participants