昇腾社区首页
中文
注册
开发者
下载
支持
积分兑换 NEW

MSPTI工具使用（Python API）

MSPTI工具Python API介绍

当前提供如下类型的API：

  • hccl _monitor：通信算子性能数据采集接口。
  • kernel_monitor：kernel算子性能数据采集接口。
  • mstx_monitor：mstx性能数据采集接口。

使用示例

以下接口详细介绍请参见MSPTI Python API参考。

  • HcclMonitor
     1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
    from mspti import (
	HcclData,
	HcclMonitor
)

def HcclParser(data : HcclData):
	print(data.kind, data.start, data.end, data.device_id, data.stream_id, data.bandwidth, data.name, data.comm_name)

monitor = HcclMonitor()
monitor.start(HcclParser)
# 用户业务逻辑
…
monitor.stop()
  • KernelMonitor
     1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
    from mspti import (
	KernelData,
	KernelMonitor
)

def KernelParser(data: KernelData):
	print(f"[Kernel Info] kernel name: {data.name}, start: {data.start}, end: {data.end}, deviceId: {data.device_id}, streamId: {data.stream_id}, type: {data.type}, correlationId: {data.correlation_id}")

monitor = KernelMonitor()
monitor.start(KernelParser)
…
# 业务逻辑
monitor.stop()
  • MstxMonitor
    from mspti import (
	MarkerData,
	MstxMonitor			  
)

def MarkerParser(data: MarkerData):
	print(f" [Mark Data Info] flag:{data.flag}, sourceKind: {data.source_kind}, timestamp: {data.timestamp}, deviceId: {data.object_id.device_id}, streamId: { data.object_id.stream_id}, processId: {data.object_id.process_id}, thread_id: {data.object_id.thread_id}" )

monitor = MstxMonitor()
monitor.start(MarkerParser)
…
# 业务逻辑
monitor.stop()