面经总结 AI infra 攒人品

算法题:
1- 快排,寻找两个正序数组的中位数,下一个排列,二叉树中的最大路径和,Path Sum III
2- 给定若干点的数轴坐标数组和固定数量的等长线段,问该线段最少要多长才能覆盖所有点
3- 前k个高频字符串,词频一样时按字典序升序排列
4- 给定初始字符串s,每次将字符串向右旋转一次,并将旋转后的字符串拼接到原字符串的末尾,每次操作都会使字符串的长度变为原来的两倍,求计算出无限扩展后的字符串中第 N个位置的字符
5- 两根手指放在26个小写字母组成的键盘上,最少移动多少距离才能敲出给定的字符串s

orch手撕题:MHA * 3,Flash Attention v1,flow matching model采样的伪代码

ai infra或算法八股:
1- flow matching模型预测的是什么,怎么理解conditional velocity (conditioned on data sample x0)
2- 如何计算QwenImage的time shift
3- 介绍Flash Attention的原理和实现思路
4- GPU matrix transpose使用shared memory的好处
5- CPU按列遍历一个行优先的矩阵相比按行遍历为什么性能会变差,具体是因为哪个性能指标变差导致的
6- weight-only量化有哪些,实现weight-only量化cuda kernel时如何优化访存,是否了解Marlin kernel
7- Megatron SP的实现方式
8- DeepSpeed ZeRO stage1和stage 2的通信量区别,论文和代码实现有没有gap
9- 多GPU通信时NVSHMEM和NVLink的区别
全部评论
这个手撕好难,看不懂
点赞 回复 分享
发布于 03-26 16:54 江西
tql
点赞 回复 分享
发布于 03-09 11:22 山东

相关推荐

给我面没招了,感觉自己好菜、面试很难,还是要多多练习1.项目拷打2.你在去部署或者训练预训练或者后训练的模型时,有没有用过一些比较底层的一些训练的调试的工具,比如说千卡的话很容易就会出NCCL timeout,如果出现 NCCL timeout,一般怎么定位和解决?3.像那种rl里面的那个MOE之类的那种的优化有去做过吗4.看您的训练经验比较丰富,而且您上线运行的推理内容之前也进行过一些什么样的优化吗?5.有没有做过 kernel级别的优化?比如用 CUTE DSL或者手写 CUDA去做 fusion这类算子融合优化,介绍一下6像底层,如果你们在做.kernel fusion,倾向于用什么方式来做7.有没有哪次你做了 fusion 结果性能反而下降的?原因是什么8.平时写 CUDA的时候,有没有关注到底层实现细节?比如你刚提到 FA2,那再往下一层,像 Hopper架构里那个 warp specialization是什么,它底层大概是怎么实现的9.试过用 Agent去生成cuda内核么,怎么去做的10.如果我把 warp specialization 去掉,只保留 tile 和 shared memory 优化,大概会损失在哪?11.怎么么判断一个 MoE 模型是真的学到了分工,而不是只是把 dense模型拆开了12.在 RL + MoE 里,有没有遇到过 reward把 routing学坏的情况?就是模型为了拿 reward,全都走某几个 expert,这种情况你当时是怎么处理的
查看11道真题和解析
点赞 评论 收藏
分享
评论
3
18
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务