岗位描述
工作职责:
1.负责自研AI芯片Tile-based DSL编译器的后端设计与开发,构建高性能代码生成链路;
2.主导编译器中端优化管道建设,基于AST重写与动态追踪混合编译策略,提升IR生成效率及目标代码质量;
3.负责编译器IR在自研芯片上的功能验证与性能调优,确保编译产出与硬件微架构的高效匹配;
4.深度参与并优化基于该DSL的高性能算子库,重点覆盖通用矩阵乘(GEMM)、分组GEMM、融合多头注意力(FMHA)等核心算子;
5.与硬件架构团队紧密协同,参与芯片微架构设计评审,从编译视角驱动硬件特性优化。
任职要求:
1.计算机相关专业本科及以上学历,精通C++与Python,具备扎实的系统软件工程能力;
2.深入掌握至少一种主流DSL编程语言(如Triton、TileLang),具备编译器层面的开发或定制经验;
3.熟悉MLIR或TVM编译框架,了解LLVM后端基础设施,具备中间表示(IR)设计与降级实战经验;
4.熟悉GPGPU体系结构及CUDA编程模型,对计算机体系结构与编译原理有深刻理解;
5.了解Agent Kernel编程范式,具备异构计算场景下的调度优化意识。
加分项:
1.有计算机领域高水平编程竞赛(如ACM-ICPC、超算竞赛等)获奖经历;
2.具备CUDA/OpenCL/Triton等算子级手工调优经验,熟悉性能分析工具(如Nsight、VTune);
3.熟悉MLIR, TVM, LLVM优先;
4.有国产AI芯片编译器或算子库实际开发经验。