小鹏汽车 社会招聘
AI存储架构设计工程师
上海 ·经验不限·学历不限
上海
薪资面议
前往 小鹏汽车 官网投递
职位描述
岗位职责:
1. AI多级存储分层架构设计
- 统筹寄存器、Shared Memory、L0/L1/L2缓存、片上全局内存、主机系统内存分层存储架构;
- 定义与设计GPGPU MMU、Cache架构;
- 定义与设计异步拷贝引擎cp.async、张量访存、批量DMA、MMIO寄存器访问通路,区分volatile/mmio特殊访存语义硬件实现;
- 针对大模型训练/推理场景优化带宽、延迟、冲突化解、读写合并、预取策略,平衡算力与存储吞吐。
2. GPU内存一致性模型(MCM)架构定义与RTL约束
- 基于Memory Consistency Model,完成GPGPU片上存储、访存流水线、多级缓存、DMA、IOMMU/SMMU整体一致性架构设计;
- 定义Memory Proxy(generic/async/tensormap/fabric)硬件通路隔离、跨Proxy同步、proxy fence屏障实现规范;
- 梳理Scope(CTA/Cluster/GPGPU/System)硬件可见域、VMID/SID/PASID地址翻译隔离逻辑;
- 负责Release/Acquire、Fence-SC、Barrier、异步访存cp.async等同步原语硬件架构设计。
2. AI加速卡虚拟化I/O存储架构设计
- 设计SR-IOV/MIG硬件虚拟化存储隔离方案,基于Stream ID、VMID实现虚拟机间DMA内存硬件隔离;
- 完成SVA共享虚拟地址、PASID子流ID、IOMMU页表翻译、Stage1/Stage2两级地址映射架构;
- 解决多虚拟机、多MIG切片并发访存冲突、数据竞争、混合尺寸访存原子性问题,输出虚拟化存储隔离规范。
3. 数据竞争、原子性与同步硬件方案落地
-输出架构spec、架构文档,指导设计与验证团队完成存储单元设计与验证。
4. 前沿架构预研
- 跟踪新一代GPU/AI芯片内存一致性、分布式存储、多卡Fabric、MIG虚拟化、统一地址空间技术;
- 针对大模型、多租户云推理、自动驾驶DRIVE场景输出存储架构预研方案与性能评估报告。
【任职要求】
任职要求:
一、硬性基础要求:
1. 本科及以上,微电子、集成电路、计算机体系结构、电子工程等相关专业;3年及以上GPU/AI芯片/高性能处理器存储架构设计经验;
2. 精通CPU/GPU内存一致性模型、缓存一致性、访存流水线、多级缓存架构,熟悉PTX/ISA内存语义优先;
3. 掌握IOMMU/SMMU、SR-IOV、MIG、VF直通、Stream ID、VMID、PASID虚拟化I/O存储隔离架构;
4. 熟悉硬件同步原语:fence、barrier、atomic RMW、release/acquire、proxy fence硬件实现;
5. 能够独立输出架构Spec、微架构文档;
6. 熟悉Verilog/SystemVerilog,RTL代码,熟悉存储子系统仿真流程。
二、专业深度加分项(核心匹配本岗位):
1. 熟悉Memory Consistency Model,理解Memory Proxy、Scope、因果序、通信序、一致性;
2. 有GPU异步访存cp.async、张量访存、NVLink多卡Fabric存储架构设计经验;
3. 做过vGPU/MIG虚拟化、多租户硬件隔离、SVA共享虚拟地址相关架构;
4. 熟悉HBM/DDR缓存、片上共享内存、存储通路差异化设计;
6. 熟悉AI大模型训练/推理、自动驾驶DRIVE计算平台存储负载特征。
三、软素质要求:
1. 良好跨团队沟通能力,可对齐编译器、驱动、RTL、验证、后端多团队;
2. 严谨细致,擅长Spec编写;
3. 具备自驱力,能独立完成架构预研、方案对比、性能瓶颈分析。
岗位职责补充(业务场景)
面向通用推理加速卡、车载自动驾驶GPU平台,负责芯片全栈存储架构,解决多线程、多虚拟机、多卡互联场景下内存可见性、硬件隔离、访存带宽瓶颈问题,支撑大模型、自动驾驶、云端虚拟化业务落地。
官网发布:2026-07-22 · 最后确认在招:2026-09-03 19:36:58 · 来源平台:feishu