线性层、Attention 和 FFN 如何实施张量并行,列并行与行并行怎样组合?

未标记

Sorry o(╥﹏╥)o,暂无参考答案。带这道题,问问AI吧~

问 AI
本题目录

相关面经

8 篇

腾讯混元

AI Infra
一面
查看原面经 →
线性层、注意力模块、FFN模块分别怎么做张量切分?列并行、行并行的切分逻辑,以及对应的通信位置。
追问
列并行切的是输入维度还是输出维度?
追问
行并行的切分方式是什么?
追问
QKV模块如何落地TP?
追问
Attention头怎么分配到不同卡上?
追问
FFN两层结构该怎么搭配切分?
追问
工程上为什么经常把列并行+行并行组合使用?

爱奇艺

AI 平台研发工程师(Infra 方向)
校招春招一面、二面、三面、四面
查看原面经 →
TP的通信次数

百度

AI Infra 实习生
实习
查看原面经 →
tp并行:基本思想,attention和gemm怎么切

寒武纪

AIGC 推理框架研发工程师
校招春招一面、二面
查看原面经 →
TP下,不同的模块如何汇总结果

快手

AI Infra
一面
查看原面经 →
大模型的一层有几个线性层?TP的时候怎么切的?这样子做的原因是什么?有什么思路优化中间的allreduce吗?
展开剩余 3 篇面经

公司未注明

AI Infra
实习
查看原面经 →
Megatron,tp是怎么切分的,mlp中第一个矩阵和第二个矩阵分别是行切还是列切,通信分别是什么算子;

科大讯飞

AI Infra
校招一面
查看原面经 →
TP原理:Tensor Parallel切分的是什么?涉及哪些通信?

字节跳动

AI Infra
实习
查看原面经 →
张量并行先列切和先行切区别

相关题目

4 题