Skip to content

Commit

Permalink
bug fix of dynamo_timed in cprofile (#128203)
Browse files Browse the repository at this point in the history
Summary:
Fixes #ISSUE_NUMBER

fb-only: "Entire Frame" was missing before this change.

Before: https://interncache-all.fbcdn.net/manifold/tlparse_reports/tree/logs/f565966006-TrainingApplication/20240527/rank_0/5_0_1/compilation_metrics_23.html
After: https://interncache-all.fbcdn.net/manifold/tlparse_reports/tree/logs/f569854578-TrainingApplication/20240606/rank_0/0_0_0/compilation_metrics_16.html

X-link: pytorch/pytorch#128203
Approved by: https://github.com/Chillee

Reviewed By: clee2000

Differential Revision: D58333931

Pulled By: dshi7

fbshipit-source-id: 50fa92c1e070732705c57c0fe914d3212b7ddbba
  • Loading branch information
dshi7 authored and facebook-github-bot committed Jun 10, 2024
1 parent 98ed700 commit aae0125
Showing 1 changed file with 0 additions and 3 deletions.
3 changes: 0 additions & 3 deletions userbenchmark/dynamo/dynamobench/_dynamo/utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -215,9 +215,6 @@ def _add_time_spent(key, phase_name, time_spent):

def dynamo_timed(original_function=None, phase_name=None, fwd_only=True):
def dynamo_timed_inner(func):
if config.cprofile:
return func

@wraps(func)
def time_wrapper(*args, **kwargs):
key = func.__qualname__
Expand Down

0 comments on commit aae0125

Please sign in to comment.