职位描述
1、负责openai triton编译器开发工作,适配自研GPU芯片;
2、芯片指令集在自研芯片上的功能验证及性能调优;
3、开发、适配和优化Triton算子库。并对Triton算子库进行性能测试与分析,提出性能改进方案;
4、与编译器及硬件团队合作,共同优化芯片架构及性能。任职要求
1、计算机相关专业,精通C++及python;
2、熟悉triton编程,有triton编译器开发经验;
3、熟悉MLIR及LLVM编译器框架;
4、熟悉GPGPU架构及cuda,精通体系结构及编译原理