facebookresearch / HolisticTraceAnalysis

A library to analyze PyTorch traces.
http://hta.readthedocs.io
MIT License
270 stars 37 forks source link

Add parent to dataframe in call_stack #135

Closed pavanky closed 4 months ago

pavanky commented 4 months ago

Summary: Add parent to the dataframe at the end of building the call_stack

1) For cpu events, look at the CallStackGraphs and find the parents. 2) For gpu events, look at the index_correlation to find the parents.

Differential Revision: D57084403

facebook-github-bot commented 4 months ago

This pull request was exported from Phabricator. Differential Revision: D57084403

facebook-github-bot commented 4 months ago

This pull request was exported from Phabricator. Differential Revision: D57084403

facebook-github-bot commented 4 months ago

This pull request was exported from Phabricator. Differential Revision: D57084403

facebook-github-bot commented 4 months ago

This pull request was exported from Phabricator. Differential Revision: D57084403

facebook-github-bot commented 4 months ago

This pull request has been merged in facebookresearch/HolisticTraceAnalysis@4e3b81d9a70d7792766382e26e4e6e1a915e2d6e.