一键安装此插件:
42plugin install wenhaochai/claude-plugins/extract-computation-graph
静态分析 PyTorch 代码,生成记录模块树与执行调度的 JSON 文件,明确计算内容、设备间数据移动及同步点。
从属套包
wenhaochai/kit/claude-plugins