MindStudio组合技,让Host Bound问题看得见、调得准

19 阅读5分钟

一、背景介绍:Host Bound 问题

在 NPU 训练和推理场景中,Host 侧(CPU)的任务下发(如算子调度、内存分配)与 Device 侧(NPU)的任务执行是异步进行的。当 Host 侧任务下发耗时超过 Device 侧任务执行耗时,Device 会因等待新任务而处于空闲状态,形成性能瓶颈,即 Host Bound 问题。

Host Bound 问题现象整体表现为 Free(NPU 空闲)占比高(注意:Free 占比高不一定是 Host Bound 问题),其细分现象包含但不限于:

  • Host 侧某算子下发耗时长

  • Host 侧出现大量长耗时空泡

  • Host 侧流水中断

Host Bound 问题的根因大多数归结为:

  • Host 侧线程抢占;

  • Host 侧某算子/函数等待计算结果返回;

  • Host 侧长时间等待资源(如锁资源),或资源未及时释放导致阻塞;

  • Host 侧下发队列满;

  • H2D 数据传输异常;

其中,比较常见的是Host侧线程抢占问题。此类问题通常通过ftrace(Linux内核内置的一种跟踪工具)来排查,但是存在较大的困难:调优人员难以将ftrace数据和Profiling对齐,联合分析CPU与NPU的协同情况。

针对这个问题,MindStudio Insight 推出了基于ftrace 的采集解析脚本,解析结果可导入Insight可视化界面,联合Profiling展示,帮助用户实现 ftrace 数据和Profiling 数据联合分析,优化Host瓶颈。

二、Host Bound 问题分析与解决思路

利用MindStudio Insight分析 Host Bound 问题时,可以根据以下思路:

1. 使用 MindStudio Insight 集群分析功能进行整体分析,初步确定是否存在下发慢卡;

2. 同步采集Profiling 和 ftrace 数据,联合分析CPU与NPU协同情况; 

3. 根据分析结果,进行针对性优化,例如绑核优化、核隔离等方式,减少进程抢占现象。

三、Host Bound优化实战案例

1 问题背景

某Linux服务器,已经发现了性能膨胀和Host侧空泡问题,但是仅仅从 Profiling数据,无法判断真实原因,缺少对于Host侧的性能分析手段。

使用MindStudio Insight的集群分析功能,看到卡3的Free占比明显过高,为下发慢卡。

图片

图1 集群分析初步确定是否存在下发慢卡

2 Profiling 结合 ftrace 分析抢占情况

2.1 Profiling 分析 Device 侧现象

通过 Profiling 数据,对Device侧分析(下图),可以很明显的看到卡3、卡5在黄色框选部分存在空泡,没有充分利用硬件性能。

图片

图2 -总体 Device 侧执行情况

以卡3为例,其Host To Device 连线坡度逐渐垂直,说明卡3 Host 侧下发任务较慢,Device侧出现空转,导致硬件资源浪费。

图片

图3-卡3 Host To Device 下发连线

经过以上分析,基本可以确认,Host Bound下发慢卡降低了NPU利用率。但此时,仅通过 Profiling 数据,已经不足够分析Host侧问题根因了。接下来需要结合 ftrace数据,分析当时CPU的行为有哪些异常。

2.2 采集 ftrace 数据

利用MindStudio Insight 提供的采集脚本 trace_record.py,采集 ftrace 数据。运行结束后在当前目录下生成 ftrace.txt 文件。然后使用转换脚本trace_convert.py,将 ftrace 数据格式转换为能够导入MindStudio Insight的格式。

以上脚本,可以在MindStudio Insight的开源代码仓中获取:

gitcode.com/Ascend/msin…

2.3 Profiling与ftrace联合分析

发现问题一:ACLThread 被 CANN 线程抢占

如下图4所示,可以看到 async_task_queue连线(用于关联Python至CANN的任务下发关系)坡度非常大,这说明Python层的下发队列中有任务,但是一直没有下发,造成大约 3.2ms 的空泡。

进一步定位问题原因。该环境已经使用过绑核,绑核策略中,对应卡4的 ACLThread 绑在 CPU 90 上。因此,将泳道名称为 CPU 90 的 ftrace 泳道置顶,区域框选后发现,卡4下发慢时,对应的 Host 侧 CPU 被dev14_sq_task 进程抢占

图片

图4-卡4有大约 3.2ms 的空泡

定位结论与解决措施

dev_sq_task 责将任务传输到 Device 上,任务优先级较高,抢占了ACLThread。在此期间,ACLThread一直处于就绪但未被调度的状态,导致了3.2ms空泡。因此,需要将 ACLThread、dev_sq_task 分别绑核。

发现问题二: ACLThread 被系统级线程systemctl抢占

如下图5所示,在两个标记(绿色旗帜和红色旗帜之间)区域,对应 CPU 被 systemctl 线程抢占并打断,导致 Host Bound 问题。图6可以看得更清楚。systemctl 线程是内核线程,主要用来做 Linux 上各种服务的调度和更新,优先级高于用户线程。

图片

图5-卡4分析系统线程抢占

图片

图6-系统线程抢占 ACLThread

解决措施

启用 /etc/systemd/system.conf  中的CPU亲和性配置属性,将系统线程绑定在某几个特定的CPU核心上,以避免对用户线程的抢占。

发现问题三:频繁唤醒休眠导致CPU空转

如下图7所示,发现慢卡主进程虽然没有被抢占,但是主进程频繁在 sched_waking 状态。

图片

图7-frace.txt 的 sched_waking 状态频繁

解决措施

使用CANN包提供的下发流水优化特性,开启二级流水优化,以避免主进程过快进入休眠状态而被频繁唤醒。可通过export TASK_QUEUE_ENABLE=2 环境变量使能该特性,详见:

www.hiascend.com/document/de…

4 优化结果

通过 Profiling 结合 ftrace 分析进程抢占和频繁调度问题,并且针对问题完成绑核、使能下发流水优化等方法,最终将 Free 占比从 18% 降低到 0.38%,Host Bound问题得到极大缓解。

图片

图8-优化前 Free 占比

image.png

图9 -优化后 Free 占比

总结

Host Bound 问题已经成为模型训练与推理部署中的高频问题,但是一直以来,调优人员都缺乏有效手段,能够同时观测 Host 侧和 Device 侧的执行情况。MindStudio 提供了一种将Profiling数据与ftrace数据联合分析的方案,打开了Host侧黑盒,让下发问题看得准,调得快。

工具链接如下

  1. MindStudio Insight可视化工具:gitcode.com/Ascend/msin…
  1. ftrace 数据采集与解析脚本:gitcode.com/Ascend/msin…
  1. Profiling采集工具:gitcode.com/Ascend/mspr…