跳到主要内容
T4 车企/自驾 元戎启行 校招 一面

元戎启行 AI Infra 校招 一面 (2)

元戎启行 AI Infra 校招一面面试真题,涵盖算子优化等方向

算子优化

编程题

  1. 实现四维张量的 Concat 操作,分别描述在每个维度上进行拼接的处理方式。
  2. 使用 CUDA 编写四维张量的 Concat Kernel,针对不同维度分别实现。
  3. 实现 LRU(Least Recently Used)缓存。