NVIDIA 嘅 Nsight 工具線新出咗一個開源 Python 接口,開發者可以直接喺 Python 程式碼入面做 GPU kernel profiling(GPU kernel 性能分析),唔使跳去獨立 profiler 介面。

呢個接口喺 Python 入面可以做幾樣嘢:

  • Profile GPU kernels:直接對 GPU 上跑緊嘅 kernel 做性能分析
  • 分析硬件指標:讀取同分析硬件層面嘅 metrics
  • 評估 cache 同時鐘控制:對 cache 行為同時鐘控制做評估
  • 量度 kernel fusion:量度 kernel fusion(將多個小 kernel 合併成一個,減少啟動開銷)嘅效果

對於要喺 Python 生態系統入面做 GPU 性能調優嘅開發者嚟講,唔使切換到另一套 profiler 環境,直接喺自己嘅 Python 腳本度就可以做呢啲分析,係一個實用嘅補充。

NVIDIA 嘅 Nsight 系列工具(包括 Nsight Systems、Nsight Compute)一直係獨立命令行或者獨立介面。今次提供 Python 接口,把呢啲能力直接接入 Python 工作流。

模姐|深度調教