We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 216b6a9 commit f98c601Copy full SHA for f98c601
torch/_dynamo/utils.py
@@ -162,12 +162,14 @@ class ReinplaceCounters:
162
# Track sizes of known not re-inplaced tensors (exclude dynamic shapes).
163
@classmethod
164
def add_missed_bytes(cls, trigger: ReInplaceTrigger, bytes: int):
165
- cls._values[f"missed_bytes_{trigger.name}"] += bytes
+ if bytes != 0:
166
+ cls._values[f"missed_bytes_{trigger.name}"] += bytes
167
168
# Track number of not re-inplaced tensors.
169
170
def add_missed_opportunities(cls, trigger: ReInplaceTrigger, count: int):
- cls._values[f"missed_tensors_{trigger}"] += count
171
+ if count != 0:
172
+ cls._values[f"missed_tensors_{trigger}"] += count
173
174
175
def clear(cls):
0 commit comments