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

未标记

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

问 AI
本题目录

相关面经

1 篇

腾讯混元

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

相关题目

4 题