直招.cv

← 返回职位列表

小鹏汽车 社会招聘

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