MSPTI工具使用（Python API）

MSPTI工具Python API介绍

当前提供如下类型的API：

• HcclMonitor：通信算子性能数据采集接口。
• KernelMonitor：kernel算子性能数据采集接口。
• MstxMonitor：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(f"[Hccl Info] kind:{data.kind}, start:{data.start}, end:{data.end}, device_id:{data.device_id}, stream_id:{data.stream_id}, bandwidth:{data.bandwidth}, name:{data.name}, comm_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

  1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17

  from mspti import ( MarkerData, MstxMonitor, RangeMarkerData ) def MarkerParser(data: MarkerData): print(f" [Mark Data Info] flag:{data.flag}, source_kind:{data.source_kind}, timestamp:{data.timestamp}, device_id:{data.object_id.device_id}, stream_id:{ data.object_id.stream_id}, process_id:{data.object_id.process_id}, thread_id:{data.object_id.thread_id}" ) def RangeParser(data: RangeMarkerData): print(f"[RangeMark Data Info] kind:{data.kind}, source_kind:{data.source_kind}, id:{data.id}, name:{data.name}, device_id:{data.object_id.device_id}, stream_id:{data.object_id.stream_id}, process_id:{data.object_id.process_id}, thread_id:{data.object_id.thread_id}, start:{data.start}, end:{data.end}") monitor = MstxMonitor() monitor.start(MstxParser,RangeParser) … # 业务逻辑 monitor.stop()