岗位描述
岗位职责:
作为一名驱动工程师,您将与软硬件团队一起解决AI加速器的使能、测试与优化工作,为深度大模型训练和推理提供高性能、高扩展性的系统软件支持。
负责AI加速器KMD驱动的整体架构设计;
设计和开发任务调度、数据搬移、同步机制、调试接口等核心驱动模块;
深度参与KMD与用户态驱动(UMD)、底层软件、和硬件接口的协同设计;
支持跨团队协作,包括架构设计、芯片验证、系统软件、客户技术支持等。
任职要求:
计算机、微电子、电子工程或相关专业本科及以上学历;
5年以上Linux内核驱动开发经验,具备硅前/硅后开发全流程经验者优先;
熟悉PCIe协议及调试流程,具备x86或ARM平台驱动移植开发经验;
了解Firmware通信机制,如Mailbox/RPC接口设计、状态机协同执行等;
有芯片Bring-up、裸机调试、FPGA平台验证或仿真平台调试经验;
熟悉寄存器规范文档、硬件模块接口协议,有软硬件协同开发经验。
加分项:
熟悉GPU、AI或图形处理芯片的系统架构,具备从零构建驱动框架并推动落地的完整项目经验。
有多任务流调度设计经验,熟悉Stream模型、数据流管理与异步执行模型;
熟悉共享内存通信、零拷贝、pinned memory等高性能数据传输机制;
了解VFIO、SR-IOV等PCIe虚拟化机制。