岗位描述
岗位描述 :
我们正在寻找对自研AI芯片编译器和高性能计算充满热情的2027届应届毕业生,加入我们的测试开发团队。你将负责OpenAI Triton、TileLang、CuTe DSL等编程框架的质量保障与测试基础设施建设,确保编译器和DSL工具链在硬件平台上的正确性、性能与稳定性。
工作职责
编译器测试框架设计与开发:设计并实现面向Triton IR、TileLang、CuTe DSL的自动化测试框架,覆盖端到端编译流程
正确性验证:编写MLIR/LLVM IR级别的单元测试与集成测试,验证编译器各级别pass的语义正确性;构建基于参考实现(NumPy/PyTorch)的数值精度对比测试
性能回归检测:搭建性能基准测试(benchmark)体系,监控kernel在不同GPU架构上的性能变化,及时发现性能回归
CI/CD流水线建设:维护和优化持续集成系统,完成自动化测试流程搭建与结果分析
测试覆盖率提升:分析代码覆盖率数据,针对未覆盖路径设计定向测试用例,持续提升编译器代码覆盖率
问题定位与协作:与开发工程师紧密协作,高效定位和复现缺陷,推动问题修复
任职要求
2027年毕业,计算机科学、软件工程、电子工程或相关专业
扎实的编程基础,熟练掌握 Python 和 C++
熟悉编译器基本原理(词法分析、语法分析、中间表示、优化pass、代码生成)
了解至少一种主流的DSL编程语言(triton、tilelang)
了解pytorch、vllm等常用AI框架
具备良好的软件测试思维,理解测试金字塔、边界值分析、等价类划分等测试方法论
熟悉Linux开发环境和Git工作流
思维敏捷、积极主动、责任心强且具备良好的良好的沟通、表达和协调能力
加分项
有MLIR/LLVM相关项目经验或课程背景
了解或使用过OpenAI Triton编写GPU kernel
熟悉TileLang或CuTe的Tile抽象与布局(Layout)概念
有编译器测试经验(如LLVM lit测试框架、FileCheck等)
地点:上海 、南京