news 2026/10/14 3:21:41

根据时戳从pytorch trace中获取kernel信息

作者头像

张小明

前端开发工程师

1.2k 24
文章封面图
根据时戳从pytorch trace中获取kernel信息

选定区域

需要使用https://ui.perfetto.dev/打开trace信息的json文件。

获取选定区域的ts信息


start_ts = 6_758_429_328_000.000
end_ts = 6_758_429_344_000.000

使用代码获取kernel信息

trace_pro.py

importjsondeftruncate_trace(input_file,output_csv,start_ts,end_ts,delimiter:str=","):withopen(input_file,'r')asf:data=json.load(f)events=data['traceEvents']# 筛选时间范围内的完整事件filtered=[]foreinevents:ife.get('ph')!='X':continueargs=e.get('args',{})if'stream'notinargs:# 关键:只留含 stream 字段的事件continuets=e.get('ts',None)iftsisNoneorts<start_tsorts>end_ts:continuefiltered.append(e)# 按时间戳排序filtered.sort(key=lambdax:x['ts'])# 写入 CSVwithopen(output_csv,'w',newline='',encoding='utf-8-sig')asf:table_name=f"name{delimiter}dur{delimiter}start{delimiter}end{delimiter}"f"cat{delimiter}pid{delimiter}tid{delimiter}"#f.write(table_name + "\n")foreinfiltered:name=e.get('name','')start=e.get('ts',0)dur=e.get('dur',0)end=start+dur cat=e.get('cat','')pid=e.get('pid','')tid=e.get('tid','')row=f"{name}{delimiter}{dur}{delimiter}{start}{delimiter}{end}{delimiter}"f"{cat}{delimiter}{pid}{delimiter}{tid}{delimiter}"f.write(row+"\n")print(f"write rows{len(filtered)}")defprocess_new():start_ts=867_295_606_000.000end_ts=867_295_626_000.000input_file='in32768_bs256-1791724311.7331705-TP-0.trace.json'output_csv='kernels_1013_256_new.csv'truncate_trace(input_file,output_csv,start_ts,end_ts,delimiter=";")defprocess_old():start_ts=6_758_429_328_000.000end_ts=6_758_429_344_000.000input_file='in32768_bs128-1789726212.8056748-TP-0.trace.json'output_csv='kernels_1013_128_old.csv'truncate_trace(input_file,output_csv,start_ts,end_ts,delimiter=";")truncate_trace(input_file,output_csv,start_ts,end_ts,delimiter=";")if__name__=="__main__":process_old()

使用excel打开处理后的csv文件

代码中使用分号(delimiter=“;”)作为分割符。
打开excel,数据,从文本/csv。

版权声明: 本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若内容造成侵权/违法违规/事实不符,请联系邮箱:809451989@qq.com进行投诉反馈,一经查实,立即删除!
网站建设 2026/10/14 3:21:11

MATLAB TCP/IP网口调用:从tcpclient连接到字节序与粘包处理

简介&#xff1a;面向MATLAB用户的TCP/IP网口调用示例程序&#xff0c;主要解决远程设备数据通信、控制系统交互等场景下的网络编程需求&#xff0c;适合有一定MATLAB基础、希望快速上手网口通信的开发者和科研人员。压缩包内共含2个文件&#xff1a;1个可直接运行的.m脚本文件…

作者头像 李华
网站建设 2026/10/14 3:21:03

Faithfulness、Context Recall…RAGAS 核心指标怎么算(检索侧)(4/13)

总目录 [第 1 章] 你的 RAG 为什么总在「一本正经地胡说」 [第 2 章] 一张图看懂 RAGAS&#xff1a;核心概念与 7 大指标 [第 3 章] 数据构建——样本、数据集与模型接入 [第 4 章] Faithfulness、Context Recall…RAGAS 核心指标怎么算 [第 5 章] 生成侧指标与指标选型&#…

作者头像 李华
网站建设 2026/10/14 3:20:39

mdeditor v2.0集成实战:Markdown编辑器部署、配置与避坑指南

简介&#xff1a;mdeditor v2.0是一款高效在线Markdown编辑器资源包&#xff0c;定位服务于程序员、内容创作者、毕业设计学生和网站搭建者&#xff0c;可用于论文排版、技术文档编写、代码片段展示以及建站后台的内容管理&#xff1b;编辑器支持实时预览、主题定制、代码高亮、…

作者头像 李华
网站建设 2026/10/14 3:20:00

【零基础学智能仿真-50】让智能体安全调用仿真工具:状态反馈、错误处理与结果核查

课程摘要 上一节&#xff0c;我们将自然语言需求整理成经过校验的仿真任务书。本节继续完成智能体工作流&#xff1a;如何按顺序调用工具、区分“工具调用成功”与“求解成功”&#xff0c;以及在许可证故障、结果缺失和物理量异常时停止流程。通过一个不连接 Abaqus 的模拟程序…

作者头像 李华