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 "Add dispatch mode testing for meta tensors and other stuff"

Browse Source 
This reverts commit c1cdb1216b97970d903a6d6e9e7d0e2b4ffaef46.

Reverted https://github.com/pytorch/pytorch/pull/77477 on behalf of https://github.com/malfet
This commit is contained in:
PyTorch MergeBot
2022-05-18 02:56:48 +00:00
parent c35bd8d423
commit 48581d74ad
16 changed files with 846 additions and 1214 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
test/test_python_dispatch.py
View File

@ -894,8 +894,8 @@ $1 = torch._ops.aten.add.Tensor($0, $0)''')
return func(*args, **kwargs)
x = torch.randn(1)
with Logger.push("A"):
with Logger.push("B"):
with push_torch_dispatch_mode(partial(Logger, "A")):
with push_torch_dispatch_mode(partial(Logger, "B")):
x + x
self.assertEqual(logs, ["B", "A"])