岗位职责
- 从事RISC-V CPU 编译器的研发工作,包括编译器架构设计、优化遍开发、应用程序性能分析调优等工作;
- 基于 MLIR + Linalg 框架,提升编译器代码块执行效率;
- 参与图优化、后端代码生成、调度策略、指令优化、流水并行化相关优化工作;
- 研究和优化 Kernel 计算性能,降低计算开销,提高吞吐率;
- 研究算子融合(Op Fusion)、自动调优(Auto-Tuning)、代码生成(CodeGen)相关优化策略;
- 与硬件团队、AI 框架团队(如 PyTorch)协作,提升 AI 编译器的支持能力; 跟踪 LLVM、Triton、TVM、XLA、TorchInductor 等 AI 编译器前沿技术,优化编译性能。
职位要求
- 计算机、自动化和电子类相关专业,本科及以上学历。
- 熟悉NPU/GPU/DSP等硬件计算单元特性,对某优化策略有深入研究。
- 熟悉深度学习编译相关工作,有丰富的MLIR/TVM/LLVM/XLA/IREE等编译器开发经验。
- 有计算机体系结构背景,处理器微架构软硬件协同优化经验的优先。
- 良好的团队沟通能力与问题解决能力。