职位描述
1. 负责Triton、TileLang 等 GPU 领域专用语言(DSL)的编译框架研发与优化;
2. 负责Triton/TileLang DSL 语法扩展与解析,支持新增 DSL 语法特性;
3. 基于 MLIR/TVM/LLVM 构建 Triton/TileLang 专用 Dialect,开发优化Pass;
4. 针对自研GPU芯片进行Triton及TileLang的编译性能优化,目标性能与手写算子一致;
5. 分析 Triton/TileLang 生成代码的性能瓶颈,通过编译层面的内存访问模式优化、指令流水线调度、自动向量化等提升代码性能;
6. 跟踪 Triton/TileLang 开源社区前沿技术,探索 DSL 编译创新方向,参与开源社区贡献。任职要求
1. 本科及以上学历,计算机、高性能计算、微电子等相关专业,5年及以上编译器开发经验;
2. 参与过 Triton/TileLang/MLIR 开源项目开发;
3. 熟悉GPGPU架构及异构编程模型,具备较强问题分析与定位能力;
4. 精通C++,CUDA及Python,具备模块化代码设计及调优能力;
5. 良好的跨团队协作能力:能与 GPU 硬件架构工程师、算法工程师及框架工程师高效沟通协作;
6. 加分项:有自研 GPU 的 Triton/TileLang 编译器后端开发经验,实现过核心算子的编译适配。