Skip to content

[dtensor] delete unused torch_function #90449

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wanchaol wants to merge 1 commit into from
Closed

[dtensor] delete unused torch_function #90449

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
17 changes: 0 additions & 17 deletions torch/distributed/_tensor/api.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -214,23 +214,6 @@ def __repr__(self):
# TODO: consider all_gather the local tensors for better debugging
return f"DTensor(local_tensor={self._local_tensor}, device_mesh={self._spec.mesh}, placements={self._spec.placements})"

@classmethod
# pyre-fixme[3]: Return type must be annotated.
# pyre-fixme[2]: Parameter must be annotated.
def __torch_function__(cls, func, types, args=(), kwargs=None):
if kwargs is None:
kwargs = {}
# if we find nn.functional name in dispatch op, dispatch to it instead,
# this allow us to override some python level behaviors that wouldn't be
# possible in __torch_dispatch__ level.
if func.__name__ in DTensor._custom_dispatch_ops:
# dispatch to the same table as the name should be different between
# torch_function and torch_dispatch
return DTensor._custom_dispatch_ops[func.__name__](*args, **kwargs)
else:
# if not, just do nothing here
return super().__torch_function__(func, types, args, kwargs)

@classmethod
# pyre-fixme[3]: Return type must be annotated.
# pyre-fixme[2]: Parameter must be annotated.
Expand Down