Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Save PyTorch profile data to CSV #7114

Merged
merged 4 commits into from
Apr 18, 2023
Merged

Conversation

kgajdamo
Copy link
Contributor

@kgajdamo kgajdamo commented Apr 4, 2023

Enable to save pytorch profiling results to csv (top 5 most time consuming ops).

  • The --write-csv argument has been modified so that it can now take three values:
    None (do not write data to csv)
    bench (write benchmark data to csv)
    prof (write pytorch profiler data to csv)
  • Added new argument --export-chrome-trace, that enables to export chrome trace file (defaults to true).

Example command to profile with pytorch and save results to csv:
python ./inference_benchmark.py --profile --write-csv prof --export-chrome-trace False

@codecov
Copy link

codecov bot commented Apr 4, 2023

Codecov Report

❗ No coverage uploaded for pull request base (master@f1f24a0). Click here to learn what that means.
The diff coverage is 41.17%.

❗ Current head 2cdd1b9 differs from pull request most recent head a47b92b. Consider uploading reports for the commit a47b92b to get more accurate results

@@            Coverage Diff            @@
##             master    #7114   +/-   ##
=========================================
  Coverage          ?   91.17%           
=========================================
  Files             ?      436           
  Lines             ?    23956           
  Branches          ?        0           
=========================================
  Hits              ?    21843           
  Misses            ?     2113           
  Partials          ?        0           
Impacted Files Coverage Δ
torch_geometric/profile/profile.py 75.00% <39.39%> (ø)
torch_geometric/profile/__init__.py 100.00% <100.00%> (ø)

📣 We’re building smart automated test selection to slash your CI/CD build times. Learn more

@rusty1s rusty1s changed the title Save pytorch profile data to csv Save PyTorch profile data to CSV Apr 18, 2023
@rusty1s rusty1s enabled auto-merge (squash) April 18, 2023 13:10
@rusty1s rusty1s merged commit 9236cb7 into pyg-team:master Apr 18, 2023
10 checks passed
@kgajdamo kgajdamo deleted the write-csv-prof branch June 2, 2023 12:02
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

None yet

3 participants