ai-infra-interview-305

第 97 题:TensorRT的dynamic shape优化技巧?optimization profile

题目

TensorRT的dynamic shape优化技巧?optimization profile


完整讲解

一、Dynamic Shape 的挑战

输入 shape 在运行时变化(如 batch、seq_len 可变)时,TensorRT 需在 build 时 知道范围或若干典型值,才能选 kernel、做融合与显存分配。若完全动态,需用 dynamic shape APIoptimization profile

二、Optimization Profile

Optimization profile:为 dynamic 维度指定 min / opt / max(如 batch=1,8,32;seq=16,128,512)。Builder 会针对 opt 做主要优化,并保证 min~max 内均可执行;Runtime 执行时若在范围内则用同一 engine,超出则需多 profile 或重 build。多个 profile 可并存(如不同 batch 段),运行时按当前 shape 选 profile。

三、优化技巧


面试要点


记忆要点

  1. Optimization profile = min/opt/max;opt 为主优化目标。
  2. 范围不过宽;多 profile 可覆盖大跨度。
  3. 固定维尽量固定。
返回模块 返回总览