跳转到主要内容

国产平台适配工程师(算子开发、华为昇腾平台)

Technology
深圳市大数据研究院
深圳-福田区, 中国¥20,000 - ¥40,000 /月2周前

职位描述

岗位职责 1、负责将依赖英伟达CUDA生态的软件包(含cuDNN、cuBLAS、NCCL等底层库的调用链路)迁移适配至国产NPU(昇腾CANN等)及国产CPU(鲲鹏等)平台,完成编译、链接、运行时兼容性修复与功能验证。 2、分析待适配软件中的计算热点与算子依赖,针对国产硬件特性完成算子替换、算子融合或自定义算子开发,保证适配后功能正确性与可接受的性能衰减。 3、配合算法团队,对涉及刚体力学、多体动力学求解的数值计算模块进行国产平台上的数值稳定性排查与性能调优,定位因浮点行为差异、并行归约顺序变化等引发的精度异常或收敛问题。 4、搭建国产平台适配的自动化测试流程,覆盖基础功能回归、数值一致性比对、多核/多卡并行正确性验证,沉淀适配案例与踩坑文档。 5、跟踪国产AI芯片软件栈(CANN、MLUware、DTK等)的版本演进,评估新版本对现有适配方案的影响,提前识别兼容性风险。 任职要求 1、计算机、人工智能、电子工程、数学或相关专业硕士及以上学历。 2、具备国产NPU或国产CPU平台的软件适配经验,至少完整参与过一个软件包从x86/CUDA环境向国产平台的迁移落地。 3、熟悉以下至少一种国产软件栈:华为CANN、昆仑芯XRT,或具备跨平台编译工具链(CMake交叉编译、容器化构建)的实战能力。 4、熟练掌握 Python 与 C/C++,具备扎实的数据结构与算法功底,能在Linux环境下独立完成依赖分析、编译调试与性能剖析。 加分项 1、理解数值计算的基本概念,对以下至少一个方向有实际接触:刚体动力学、多体动力学、线性方程组求解。不要求独立推导算法,但要求能读懂求解器代码中的数值流程,判断适配后结果是否“算对了”。 2、熟悉CUDA编程模型的基本概念(kernel、线程层次、共享内存、归约操作),能识别代码中与CUDA绑定的部分并评估迁移方案。

Keywords
Python

对这个职位感兴趣吗?