Update README.md

README.md

@@ -14,6 +14,27 @@
     - [Llama预训练](###Llama预训练)
 - [参考](#参考)
 
+# 更新日志
+
+2024.12.16适配了torch prof
+
+使用方法: 启动脚本中添加下列参数, 即可采集对应的prof信息
+```bash
+PROFILE_ARGS=(
+    --profile # 开启profile
+    --profile-step-start 4 # skip前3个iter, warm第4个iter
+    --profile-step-end 5 # 采集第5个iter
+    --use-pytorch-profiler # 使用torch prof
+    --profile-ranks 0 3 # 采集全局rank 第0和3
+    --profile-dir ./prof_data # prof文件的保存目录
+)
+APP="... \
+    ${PROFILE_ARGS[@]} \
+"
+${APP}
+```
+
+
 # 环境配置
 1. 安装基础依赖包
 <pre>