frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-21 13:44:15 +08:00
Code Issues Packages Projects Releases Wiki Activity

Revert "Remove split functional wrapper (#74727)"

Browse Source 
This reverts commit a58876ace78df1cfeb136cad592487f34d7e02f1.

Reverted https://github.com/pytorch/pytorch/pull/74727 on behalf of https://github.com/seemethere due to Fails internal use cases, might extend out to external use cases as well. Need to assess overall impact of this change more widely
This commit is contained in:
PyTorch MergeBot
2022-08-10 19:45:23 +00:00
parent 651c13166c
commit f534b2c627
11 changed files with 83 additions and 92 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_python_dispatch.py
View File

@ -393,13 +393,11 @@ $5 = torch._ops.aten.clone.default($4, memory_format=torch.contiguous_format)'''
def test_list_ret(self) -> None:
# test all sequence types are permissible returns
for list_type in (list, tuple):
class A(torch.Tensor):
class A(torch._C._TensorBase):
@staticmethod
def __new__(cls, elem):
return torch.Tensor._make_subclass(cls, elem, elem.requires_grad)
__torch_function__ = torch._C._disabled_torch_function_impl
@classmethod
def __torch_dispatch__(cls, func, types, args=(), kwargs=None):
if func.overloadpacket == torch.ops.aten.split: