关注
手撕看起来不是很难
查看原帖
点赞 评论
相关推荐
03-03 10:35
门头沟学院 机器学习 在改简历的大卫很认真:1. 训练时间估算
时间 ∝ 总FLOPs / (算力FP8/FP16 × 利用率)
总FLOPs ≈ 6 × 参数 × Token数
再除以单卡算力、并行数、利用率,就是时长。
2. Prefill / Decode 优化
- Prefill:长序列并行、FlashAttention、分块、算子融合、向量化
- Decode:KV cache、连续Batch、PagedAttention、Speculative Decoding
3. Two-batch overlap
- 同时跑两个微批次,计算/通信重叠
- 负优化:计算太快、通信太小、显存不够、调度 overhead 盖过收益
4. Megatron-LM 通信优化
TP+PP+DP三维并行;
重叠通信计算;
分桶AllReduce;
激活重计算;
减少跨节点通信量。
5. PD分离(PP+DP)为什么要KV transfer
为了流水线+数据并行一起用,提高吞吐;
虽然有KV拷贝成本,但整体能堆更大Batch、更高利用率。
6. MuON / AdamW 不能混用
优化器状态、动量、更新逻辑完全不兼容;
预训练用一个,微调换另一个,状态不匹配,直接训崩/不收敛。
7. 跨SM PD / AF 分离
核心:拆任务、提高 occupancy、隐藏延迟;
PD:并行度拆分更干净;
AF:聚焦注意力/前馈访存与计算重叠;
看模型结构、Kernel瓶颈选。
8. CUDA Global / Shared 注意
- Global:对齐、连续访问、合并访存、少随机读、用L2缓存
- Shared:防bank conflict、分块复用、同步_sync、容量控制
9. DeepSeek-V3 优化点
高效架构+长序列支持;
优化Attention/FFN计算;
更好的并行策略;
推理侧低延迟高吞吐。
10. DeepSeek-DSA/NSA/MoBA
- DSA:领域稀疏注意力
- NSA:神经稀疏注意力
- MoBA:混合专家注意力
核心都是稀疏化、省计算、扩容量。
11. NCCL 原语 & AllReduce 次数
原语:Send/Recv、Broadcast、Reduce、AllGather、ReduceScatter、AllReduce
Ring AllReduce:2次(ReduceScatter + AllGather)。
12. 小数据量 NVSHMEM vs Ring AllReduce
少了Ring多跳延迟;
直接GPU-GPU读+本地Reduce;
小数据下延迟更低、更简单。
13. 超长序列训练并行
TP+PP+CP(上下文并行)+ ZeRO;
分块Attention、分块FFN;
激活重算+KV复用;
控制通信量。
14. Ampere → Hopper 算子升级
- 用TF32、FP8、新指令
- 适配TMA、异步拷贝
- 优化Warp调度、 occupancy
- 重写Attention/MatMul,用Hopper专属优化
查看14道真题和解析 点赞 评论 收藏
分享
牛客热帖
更多
正在热议
更多
# 大厂实习和小厂实习最大的区别是什么? #
8387次浏览 57人参与
# 你都用vibe coding做过什么? #
1106次浏览 44人参与
# 开放七大实习专项,百度暑期实习值得冲吗 #
20858次浏览 396人参与
# 如果人生可以debug你会改哪一行? #
1422次浏览 40人参与
# Vibe Coding 会干掉初级岗位吗? #
2434次浏览 54人参与
# 做完笔试后你收到面试了吗? #
2379次浏览 34人参与
# 厦门银行科技岗值不值得投 #
10847次浏览 271人参与
# AI Coding实战技巧 #
863次浏览 28人参与
# 你见过哪些招聘隐形歧视? #
1431次浏览 22人参与
# 牛友の3月总结 #
5780次浏览 54人参与
# 你现在一天AI几次? #
901次浏览 36人参与
# 哪些公司真双非友好? #
70998次浏览 305人参与
# 找AI工作可以去哪些公司? #
21811次浏览 973人参与
# 从事AI岗需要掌握哪些技术栈? #
17745次浏览 1085人参与
# 4399求职进展汇总 #
58874次浏览 242人参与
# 金三银四,你的春招进行到哪个阶段了? #
25975次浏览 302人参与
# 如果可以选,你最想从事什么工作 #
763985次浏览 4895人参与
# 你做过最难的笔试是哪家公司 #
53668次浏览 980人参与
# 你觉得技术面多长时间合理? #
176589次浏览 1204人参与
# AI面会问哪些问题? #
40988次浏览 1298人参与
# 说说你知道的学历厂 #
391231次浏览 1381人参与
# 中国电信笔试 #
33601次浏览 308人参与