Skip to content

Commit f98c601

Browse files
laithsakkapytorchmergebot
authored andcommitted
Avoid logging zeros (pytorch#139968)
Summary: title Test Plan: NA Differential Revision: D65582953 Pull Request resolved: pytorch#139968 Approved by: https://github.com/zou3519
1 parent 216b6a9 commit f98c601

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

torch/_dynamo/utils.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -162,12 +162,14 @@ class ReinplaceCounters:
162162
# Track sizes of known not re-inplaced tensors (exclude dynamic shapes).
163163
@classmethod
164164
def add_missed_bytes(cls, trigger: ReInplaceTrigger, bytes: int):
165-
cls._values[f"missed_bytes_{trigger.name}"] += bytes
165+
if bytes != 0:
166+
cls._values[f"missed_bytes_{trigger.name}"] += bytes
166167

167168
# Track number of not re-inplaced tensors.
168169
@classmethod
169170
def add_missed_opportunities(cls, trigger: ReInplaceTrigger, count: int):
170-
cls._values[f"missed_tensors_{trigger}"] += count
171+
if count != 0:
172+
cls._values[f"missed_tensors_{trigger}"] += count
171173

172174
@classmethod
173175
def clear(cls):

0 commit comments

Comments
 (0)