mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-21 05:34:18 +08:00
Revert "[dynamo] Special path for cloning of torch dispatch tensors (#164081)"
This reverts commit 811c693c49f7cd3da2ea174955d12f2f8780bd46. Reverted https://github.com/pytorch/pytorch/pull/164081 on behalf of https://github.com/yangw-dev due to broke internal tests ([comment](https://github.com/pytorch/pytorch/pull/164084#issuecomment-3348862668))
This commit is contained in:
@ -87,7 +87,6 @@ from torch._utils_internal import (
|
||||
from torch.fx._utils import _format_graph_code, lazy_format_graph_code
|
||||
from torch.monitor import _WaitCounter
|
||||
from torch.nn.modules.lazy import LazyModuleMixin
|
||||
from torch.utils._python_dispatch import is_traceable_wrapper_subclass
|
||||
from torch.utils._triton import has_triton, has_triton_package
|
||||
from torch.utils.hooks import RemovableHandle
|
||||
|
||||
@ -2147,10 +2146,6 @@ def clone_input(
|
||||
x.shape,
|
||||
layout=x.layout,
|
||||
)
|
||||
elif is_traceable_wrapper_subclass(x):
|
||||
# Questionable - but this is required to not fail executorch related
|
||||
# torchao tests.
|
||||
return torch_clone(x)
|
||||
|
||||
needed_size = sum(
|
||||
(shape - 1) * stride for shape, stride in zip(x.size(), x.stride())
|
||||
|
Reference in New Issue
Block a user