#632·viztracer

在 PyTorch 分析器和 DataLoader 的多进程模式下使用 VizTracer --log_torch 时,如果 kineto_results 为 None,则在退出时会引发错误和 segfault。

作者: PEKEW创建于 2025年10月29日更新于 2025年11月2日

在使用 VizTracer 的 --log_torch 功能来采样 PyTorch 代码(具有多个工作进程和大型 HDF5 文件(60GB+)时,程序尝试在退出时导出 torch profiler 跟踪信息,导致出现 AttributeError/RuntimeError。最终,进程发生内核错误并退出。看来 VizTracer 无法稳健地处理 profiler 在退出时未正确生成 kineto 结果(kineto_results 为 None)的场景。

内容来源: gaogaotiantian/viztracer