• 教程 >
  • 整体跟踪分析简介
快捷方式

整体跟踪分析简介

作者: Anupam Bhatnagar

在本教程中,我们将演示如何使用整体跟踪分析 (HTA) 来分析来自分布式训练作业的跟踪信息。要开始使用,请按照以下步骤操作。

安装 HTA

我们建议使用 Conda 环境来安装 HTA。要安装 Anaconda,请参阅Anaconda 官方文档

  1. 使用 pip 安装 HTA

    pip install HolisticTraceAnalysis
    
  2. (可选且推荐) 设置 Conda 环境

    # create the environment env_name
    conda create -n env_name
    
    # activate the environment
    conda activate env_name
    
    # When you are done, deactivate the environment by running ``conda deactivate``
    

入门

启动 Jupyter Notebook 并将 trace_dir 变量设置为跟踪信息的存储位置。

from hta.trace_analysis import TraceAnalysis
trace_dir = "/path/to/folder/with/traces"
analyzer = TraceAnalysis(trace_dir=trace_dir)

时间细分

为了有效利用 GPU,了解它们在特定作业中如何分配时间至关重要。它们主要用于计算、通信、内存事件,还是处于空闲状态?时间细分功能提供了对这三类时间消耗的详细分析。

  • 空闲时间 - GPU 处于空闲状态。

  • 计算时间 - GPU 用于矩阵乘法或向量运算。

  • 非计算时间 - GPU 用于通信或内存事件。

为了实现高训练效率,代码应最大化计算时间并最小化空闲时间和非计算时间。以下函数生成一个数据框,其中提供了每个进程的时间使用情况的详细细分。

analyzer = TraceAnalysis(trace_dir = "/path/to/trace/folder")
time_spent_df = analyzer.get_temporal_breakdown()
../_images/temporal_breakdown_df.png

当在 get_temporal_breakdown 函数中将 visualize 参数设置为 True 时,它还会生成一个条形图,表示按等级细分的分布。

../_images/temporal_breakdown_plot.png

空闲时间细分

深入了解 GPU 处于空闲状态的时间量及其背后的原因,可以帮助指导优化策略。当 GPU 上没有内核运行时,则认为该 GPU 处于空闲状态。我们开发了一种算法,将 Idle 时间分为三个不同的类别

  • 主机等待: 指的是由于 CPU 未能足够快地将内核入队以保持 GPU 充分利用而导致的 GPU 空闲时间。可以通过检查导致速度下降的 CPU 操作符、增加批次大小和应用操作符融合来解决这些类型的效率低下问题。

  • 内核等待: 这指的是在 GPU 上启动连续内核相关的短暂开销。可以通过使用 CUDA 图优化来最大程度地减少归因于此类别的空闲时间。

  • 其他等待: 此类别包括由于信息不足而无法当前归因的空闲时间。可能的原因包括使用 CUDA 事件在 CUDA 流之间进行同步以及启动内核的延迟。

主机等待时间可以解释为 GPU 由于 CPU 而暂停的时间。为了将空闲时间归因于内核等待,我们使用以下启发式方法

连续内核之间的间隔 < 阈值

默认阈值是 30 纳秒,可以使用 consecutive_kernel_delay 参数进行配置。默认情况下,仅针对等级 0 计算空闲时间细分。为了计算其他等级的细分,请在 get_idle_time_breakdown 函数中使用 ranks 参数。空闲时间细分可以按如下方式生成

analyzer = TraceAnalysis(trace_dir = "/path/to/trace/folder")
idle_time_df = analyzer.get_idle_time_breakdown()
../_images/idle_time_breakdown_percentage.png

该函数返回一个数据帧元组。第一个数据帧包含每个等级每个流上按类别划分的空闲时间。

../_images/idle_time.png

show_idle_interval_stats 设置为 True 时,将生成第二个数据帧。它包含每个等级每个流上空闲时间的汇总统计信息。

../_images/idle_time_summary.png

提示

默认情况下,空闲时间细分显示每个空闲时间类别的百分比。将 visualize_pctg 参数设置为 False,该函数将在 y 轴上呈现绝对时间。

内核细分

内核细分功能会细分每个内核类型(例如通信 (COMM)、计算 (COMP) 和内存 (MEM))所花费的时间,涵盖所有等级,并显示在每个类别中花费的时间比例。以下是饼图中每个类别所占时间的百分比

../_images/kernel_type_breakdown.png

内核细分可以按如下方式计算

analyzer = TraceAnalysis(trace_dir = "/path/to/trace/folder")
kernel_type_metrics_df, kernel_metrics_df = analyzer.get_gpu_kernel_breakdown()

函数返回的第一个数据帧包含用于生成饼图的原始值。

内核持续时间分布

get_gpu_kernel_breakdown 返回的第二个数据帧包含每个内核的持续时间汇总统计信息。特别是,这包括每个等级上每个内核的计数、最小值、最大值、平均值、标准差、总和和内核类型。

../_images/kernel_metrics_df.png

使用此数据,HTA 创建了许多可视化效果来识别性能瓶颈。

  1. 每个等级每个内核类型的顶级内核的饼图。

  2. 每个顶级内核和每个内核类型的平均持续时间(跨所有等级)的条形图。

../_images/pie_charts.png

提示

所有图像均使用 plotly 生成。将鼠标悬停在图形上会在右上方显示模式栏,用户可以使用该栏放大、平移、选择和下载图形。

上面的饼图显示了前 5 个计算、通信和内存内核。为每个等级生成类似的饼图。可以使用传递给 get_gpu_kernel_breakdown 函数的 num_kernels 参数将饼图配置为显示前 k 个内核。此外,duration_ratio 参数可用于调整需要分析的时间百分比。如果同时指定了 num_kernelsduration_ratio,则 num_kernels 优先。

../_images/comm_across_ranks.png

上面的条形图显示了 NCCL AllReduce 内核在所有等级上的平均持续时间。黑线表示每个等级上的最小和最大时间。

警告

在使用 jupyter-lab 时,将“image_renderer”参数值设置为“jupyterlab”,否则图形将不会在笔记本中呈现。

有关此功能的详细演练,请参阅存储库示例文件夹中的 gpu_kernel_breakdown 笔记本

通信计算重叠

在分布式训练中,大量时间花费在 GPU 之间的通信和同步事件上。为了实现高 GPU 效率(例如 TFLOPS/GPU),保持 GPU 超额订阅计算内核至关重要。换句话说,GPU 不应因未解决的数据依赖关系而被阻塞。衡量计算在多大程度上被数据依赖关系阻塞的一种方法是计算通信计算重叠。如果通信事件与计算事件重叠,则会观察到更高的 GPU 效率。缺少通信和计算重叠会导致 GPU 处于空闲状态,从而导致效率低下。总而言之,期望较高的通信计算重叠。为了计算每个等级的重叠百分比,我们测量以下比率

(在通信期间花费在计算上的时间)/(花费在通信上的时间)

通信计算重叠可以按如下方式计算

analyzer = TraceAnalysis(trace_dir = "/path/to/trace/folder")
overlap_df = analyzer.get_comm_comp_overlap()

该函数返回一个数据帧,其中包含每个等级的重叠百分比。

../_images/overlap_df.png

visualize 参数设置为 True 时,get_comm_comp_overlap 函数还会生成一个条形图,表示按等级划分的重叠。

../_images/overlap_plot.png

增强型计数器

内存带宽和队列长度计数器

内存带宽计数器测量在通过内存复制 (memcpy) 和内存设置 (memset) 事件将数据从 H2D、D2H 和 D2D 复制时使用的内存复制带宽。HTA 还计算每个 CUDA 流上未完成操作的数量。我们将其称为队列长度。当流上的队列长度为 1024 或更大时,无法在该流上调度新事件,并且 CPU 将停滞,直到 GPU 流上的事件处理完毕。

generate_trace_with_counters API 输出一个新的跟踪文件,其中包含内存带宽和队列长度计数器。新跟踪文件包含指示 memcpy/memset 操作使用的内存带宽的轨迹,以及每个流上队列长度的轨迹。默认情况下,这些计数器是使用等级 0 跟踪文件生成的,并且新文件在其名称中包含后缀 _with_counters。用户可以选择通过在 generate_trace_with_counters API 中使用 ranks 参数为多个等级生成计数器。

analyzer = TraceAnalysis(trace_dir = "/path/to/trace/folder")
analyzer.generate_trace_with_counters()

包含增强型计数器的生成跟踪文件的屏幕截图。

../_images/mem_bandwidth_queue_length.png

HTA 还提供内存复制带宽和队列长度计数器的摘要,以及使用以下 API 对代码的分析部分进行计数器的时间序列

要查看摘要和时间序列,请使用

# generate summary
mem_bw_summary = analyzer.get_memory_bw_summary()
queue_len_summary = analyzer.get_queue_length_summary()

# get time series
mem_bw_series = analyzer.get_memory_bw_time_series()
queue_len_series = analyzer.get_queue_length_series()

摘要包含计数、最小值、最大值、平均值、标准差、第 25 个百分位数、第 50 个百分位数和第 75 个百分位数。

../_images/queue_length_summary.png

时间序列仅包含值发生变化的点。观察到某个值后,时间序列将保持不变,直到下一次更新。内存带宽和队列长度时间序列函数返回一个字典,其键是等级,值是该等级的时间序列。默认情况下,仅针对等级 0 计算时间序列。

CUDA 内核启动统计信息

../_images/cuda_kernel_launch.png

对于在 GPU 上启动的每个事件,CPU 上都有一个相应的调度事件,例如 CudaLaunchKernelCudaMemcpyAsyncCudaMemsetAsync。这些事件通过跟踪中的一个公共相关 ID 链接 - 请参阅上图。此功能计算 CPU 运行时事件的持续时间、其相应的 GPU 内核和启动延迟,例如,GPU 内核开始和 CPU 操作符结束之间的差异。内核启动信息可以按如下方式生成

analyzer = TraceAnalysis(trace_dir="/path/to/trace/dir")
kernel_info_df = analyzer.get_cuda_kernel_launch_stats()

下面给出了生成数据帧的屏幕截图。

../_images/cuda_kernel_launch_stats.png

CPU 操作、GPU 内核的持续时间以及启动延迟使我们能够找到以下内容

  • 短 GPU 内核 - 持续时间小于相应 CPU 运行时事件的 GPU 内核。

  • 运行时事件异常值 - 持续时间过长的 CPU 运行时事件。

  • 启动延迟异常值 - 调度时间过长的 GPU 内核。

HTA 为上述三个类别中的每一个生成分布图。

短 GPU 内核

通常,CPU 端的启动时间范围为 5-20 微秒。在某些情况下,GPU 执行时间低于启动时间本身。下图帮助我们找到代码中此类实例发生的频率。

../_images/short_gpu_kernels.png

运行时事件异常值

运行时异常值取决于用于对异常值进行分类的截止值,因此 get_cuda_kernel_launch_stats API 提供了 runtime_cutoff 参数来配置该值。

../_images/runtime_outliers.png

启动延迟异常值

启动延迟异常值取决于用于对异常值进行分类的截止值,因此 get_cuda_kernel_launch_stats API 提供了 launch_delay_cutoff 参数来配置该值。

../_images/launch_delay_outliers.png

结论

在本教程中,您学习了如何安装和使用 HTA,这是一种性能工具,可让您分析分布式训练工作流中的瓶颈。要了解如何使用 HTA 工具执行跟踪差异分析,请参阅 使用整体跟踪分析进行跟踪差异

文档

访问 PyTorch 的全面开发者文档

查看文档

教程

获取针对初学者和高级开发人员的深入教程

查看教程

资源

查找开发资源并获得问题的解答

查看资源