岗位描述
岗位介绍
我们正在建设面向自研 GPGPU 的软件栈,招聘编译器研发工程师及技术专家,参与对标 NVIDIA CUDA 编译工具链(nvcc)及 PTX 中间表示相关能力的研发,打通从 GPU 编程语言、编译优化到目标指令生成的完整链路。
你将与 GPU 架构、驱动、运行时及计算库团队协作,提升编译器的正确性、程序兼容性与代码执行效率,支持 AI 和高性能计算工作负载。
岗位职责
参与 GPGPU 编译器架构设计与开发,承担前端、中间表示、优化 Pass 或后端中的一个或多个模块。
支持 CUDA C/C++ 等 GPU 编程语言的解析、语义分析和设备代码编译,完善主机端与设备端代码的编译及链接流程。
参与 PTX 相关适配,或自研 GPU 中间表示的设计与实现,支持线程组织、地址空间、同步及原子操作等并行执行语义。
面向目标 GPU 指令集开发和优化编译器后端,包括指令选择、寄存器分配、指令调度及机器码生成。
围绕典型 AI / HPC 算子开展性能分析,优化访存、控制流、寄存器使用及专用计算指令生成。
建设编译器回归测试、正确性验证和性能评测体系,定位并解决编译错误、执行异常与性能退化问题。
与硬件架构及系统软件团队协作,推动指令集、编译器与运行时接口的协同设计。
任职要求
计算机、软件工程、电子工程等相关专业本科及以上学历,具备扎实的数据结构、算法和计算机体系结构基础。
熟练掌握 C/C++,具备复杂系统软件的开发、调试及问题定位能力。
理解编译原理,熟悉 LLVM / Clang、MLIR、GCC 等至少一种编译框架,具有实际模块开发或优化经验。
熟悉以下至少一个方向:
编译器前端、类型系统、语义分析及语言扩展;
中间表示、数据流分析、循环优化及优化 Pass 开发;
编译器后端、指令选择、寄存器分配及指令调度。
理解 GPU 的 SIMT / SIMD 执行方式、存储层次及并行编程基本概念,能够结合生成代码分析程序性能。
具备良好的技术沟通和跨团队协作能力,能够独立推进问题分析、方案实现与验证。
加分项
有 CUDA、PTX、NVVM、LLVM NVPTX / AMDGPU 后端或其他 GPU 编译器相关经验。
有自研芯片编译器、指令集适配、工具链移植或软件兼容性建设经验。
熟悉 GPU 寄存器压力、Occupancy、访存合并、分支发散等性能问题。
有 Tensor Core / 矩阵计算指令、向量化或高性能算子优化经验。
有 Triton、TVM、MLIR 、Cutlass 等相关开发经验,或参与过编译器开源项目。
专家岗位额外要求
具备编译器关键模块或完整工具链的架构设计与交付经验,能够制定技术路线,组织解决兼容性、稳定性及性能方面的核心问题,并指导团队持续迭代。
欢迎具有 CPU、DSP、NPU 等编译器经验,并希望深入 GPU 编译方向的候选人。