数据转换功能

简介

本功能提供了将dump数据转换为numpy(.npy)或PyTorch tensor(.pt)格式文件的能力。支持将ATB场景的dump数据(.bin文件)或adump数据进行转换,便于后续的数据分析和处理。

使用前准备

环境准备

  • 安装msProbe工具,详情请参考《msProbe安装指南》。
  • 当指定转换格式为PyTorch tensor(.pt)时,需安装PyTorch。
  • 转换adump数据时需要确保安装配套版本的CANN Toolkit开发套件包并配置CANN环境变量,具体请参见CANN快速安装

约束

  • 支持.bin格式的ATB的dump数据转换。
  • 支持adump的数据转换。

数据转换功能介绍

功能说明

将dump数据转换为numpy(.npy)或PyTorch tensor(.pt)格式文件。

命令格式

msprobe parse -d <dump_path> [-t <type>] [-o <output_path>]

参数说明

参数名 可选/必选 描述
-d或--dump_path 必选 待转换的文件路径或目录路径。支持单个文件或目录输入:
• 单个文件:直接指定文件路径,须指定到文件名。
• 目录:指定dump文件所在目录。
-t或--type 可选 输出文件类型,支持以下两种格式:
• npy:输出为numpy(.npy)格式文件。
• pt:输出为PyTorch tensor(.pt)格式文件。
默认值为pt。
-o或--output_path 可选 输出文件路径,默认为当前路径下的output文件夹。

使用示例(转换单个dump文件)

msprobe parse -d /path/to/dump_file -o /path/to/output

使用示例(转换整个目录下的dump文件)

msprobe parse -d /path/to/dump_file_directory -o /path/to/output

输出说明

上述示例执行完成后,在--output_path参数指定路径下生成--type参数指定的格式文件。--dump_path参数指定为单个文件时只转换单个文件;参数指定为目录时,转换该目录下的所有文件。