Skip to content

Commit

Permalink
Make dynamo bridge call the new function.
Browse files Browse the repository at this point in the history
  • Loading branch information
ysiraichi committed May 24, 2024
1 parent e8add16 commit a70a8f2
Showing 1 changed file with 1 addition and 9 deletions.
10 changes: 1 addition & 9 deletions torch_xla/core/dynamo_bridge.py
Original file line number Diff line number Diff line change
Expand Up @@ -100,15 +100,7 @@ def __call__(self, args):


def get_fallback_ops():
fallback_ops = []
for opname in metrics.counter_names():
if "aten::" not in opname:
continue
val = int(metrics.counter_value(opname))
if val > 0:
fallback_ops.append(f"{opname}={val}")

return fallback_ops
return metrics.executed_fallback_ops()


# Checks that all input args that are tensors are on the same device.
Expand Down

0 comments on commit a70a8f2

Please sign in to comment.