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的区别