Remove unnecessary tree_map_only (#121052)
Summary:
Reduces the torch.compile(backend="eager") for this code by 1-2 seconds.
~~~
def fn(x):
for _ in range(10000):
# x = torch.sin(x)
x = torch.ops.aten.sin(x)
# x = sin(x)
return x
~~~
X-link: https://github.com/pytorch/pytorch/pull/121052
Approved by: https://github.com/jansel
ghstack dependencies: #121053
Reviewed By: izaitsevfb
Differential Revision: D54472072
Pulled By: anijain2305
fbshipit-source-id: fe0a1b018c8dc23160f77dd1d54be7914a3f9df5
