Skip to content

Fix/quant bugfix - #14

Closed
laksjdf wants to merge 23 commits into
jiashaokun-1:mainfrom
laksjdf:fix/quant_bugfix
Closed

Fix/quant bugfix#14
laksjdf wants to merge 23 commits into
jiashaokun-1:mainfrom
laksjdf:fix/quant_bugfix

Conversation

@laksjdf

@laksjdf laksjdf commented May 25, 2026

Copy link
Copy Markdown
Contributor

修复 expected_input_dtype() 将 MoE 内部 expert_agg add 误判为 residual boundary 的问题;expert_agg 现在保持 MoE activation dtype,只在真正的 residual2 边界插入 FP8->BF16 cast。
修复 YAML/model config 中 attn_weight_dtype, shared_expert_weight_dtype, attn_grad_dtype, shared_expert_grad_dtype 未传入 ModelSpec 的问题。
修复 memory_breakdown() 中 shared expert weight/grad dtype 被并入全局 dtype 统计的问题;routed/shared expert 现在分别按各自 dtype 计入权重和梯度显存。

laksjdf and others added 23 commits May 21, 2026 17:45
pp_p2p统计口径问题(掩盖、bwd)
寻优脚本优化,在format_result的时候增加缓存
支持pp开低精度、开overlap
支持pp开低精度、开overlap
Fix DualPipeV VPP scheduling accounting
1.fix dsv4 PARAM_SCOPE miss 2.add xlsx test
寻优允许pp=1情况,增加hidden限制
…ry 的问题;expert_agg 现在保持 MoE activation dtype,只在真正的 residual2 边界插入 FP8->BF16 cast。

修复 YAML/model config 中 attn_weight_dtype, shared_expert_weight_dtype, attn_grad_dtype, shared_expert_grad_dtype 未传入 ModelSpec 的问题。
修复 memory_breakdown() 中 shared expert weight/grad dtype 被并入全局 dtype 统计的问题;routed/shared expert 现在分别按各自 dtype 计入权重和梯度显存。
@laksjdf laksjdf closed this May 25, 2026
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