# ms_runtime_memory_tool

**Repository Path**: reku1997/ms_runtime_memory_tool

## Basic Information

- **Project Name**: ms_runtime_memory_tool
- **Description**: mindspore runtime device 内存分析工具
- **Primary Language**: Python
- **License**: MIT
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 1
- **Forks**: 2
- **Created**: 2023-08-04
- **Last Updated**: 2025-07-03

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# MindSpore大模型runtime内存分析工具

## 本工具为非官方工具

## 运行方式
1. 设置环境变量 `MS_KERNEL_LAUNCH_SKIP`=all和`MS_MEMORY_STATISTIC`=1，正常启动训练脚本，保存日志为 a.log
2. `grep 'Need Profile Memory' -a a.log >b.log`，得到需要分析的日志
3. `python3 analysis.py b.log`，得到分析结果

## 分析结果

1. 以时间为横轴，内存占用为纵轴，绘制内存占用曲线 memory.png
2. 每个内存块的生命周期、大小、传递关系等信息 blocks.csv
3. actor的执行序、流情况 actor.csv