Skip to content

AI Infra 面试基础:Notion 总结归档 ​

基础总览 · 外部笔记来源

《秋招面试总结》

包括 CUDA 算子与 Tensor 内存管理,不包含 LeetCode 类算法题。

按主题复习 ​

主题正式笔记
指针、引用、sizeof、虚函数、析构与锁C++ 对象、生命周期与并发
内存层次、合并访问、bank conflict、Stream 与 EventCUDA 内存与同步
Tensor allocator、切分、合并与异步生命周期GPU 内存池设计
RMSNorm、行 Softmax、Online Softmax归一化与在线归约
Kernel 启动、warp 调度、TMA、Triton lowering、排错GPU 执行与编译
FP32、FP16、BF16 与混合精度训练数值格式
Ring Attention、Ulysses、归一化与通信重叠分布式 Attention 基础
Prefill、Decode、Grouped GEMM、Prefix Caching推理的计算与访存成本

常见误解 ​

误解正确理解
用 sizeof(ptr) / sizeof(ref) 回答 sizeof(*ptr) / sizeof(&ref)必须按原表达式判断操作数类型,见 C++ 笔记
非虚析构只会内存泄漏常规情况下经非虚析构基类指针删除派生对象属于未定义行为
RMSNorm 用一次 shuffle 向整个 block 广播shuffle 只在 warp 内交换;第二级归约还存在参与 mask 与写竞争问题
内存池扩容后递归 alloc持有非递归互斥锁时再次加锁会死锁;GPU 释放还需要处理 stream 生命周期
Online Softmax 示例已实现原代码省略了 sum 归约、统计更新和输出写回,是未完成草稿
Ulysses = All-Gather KV + Reduce-Scatter 输出Ulysses 用两次 All-to-All 转换 sequence/head 分片;原方案还混淆了 token 与 head 对应关系
Jagged Sequence 等同任意 custom mask变长 batch、causal 对齐、任意稀疏 mask 是不同接口能力
Warp 有 Active / Eligible / Stalled / Done 四个互斥状态NCU 的 Active 包含 Eligible,不能当作四态状态机

相关主题 ​

记录原理、连接知识、积累实践。