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

Fake Tensor Part 1

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/77969

Approved by: https://github.com/ezyang
This commit is contained in:
Elias Ellison
2022-05-31 07:02:17 -07:00
committed by PyTorch MergeBot
parent d136852bda
commit 678213ead2
13 changed files with 273 additions and 36 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_python_dispatch.py
View File

@ -7,8 +7,9 @@ from torch.library import Library
from torch.cuda.jiterator import _create_jit_fn
import unittest
from torch.testing._internal.common_utils import TestCase, run_tests, TEST_WITH_ROCM, IS_WINDOWS
from torch.utils._mode_utils import no_dispatch
from torch.testing._internal.logging_tensor import LoggingTensor, LoggingTensorReentrant, LoggingTensorMode, \
log_input, capture_logs, no_dispatch, capture_logs_with_logging_tensor_mode
log_input, capture_logs, capture_logs_with_logging_tensor_mode
from torch.utils._pytree import tree_map
from torch.utils._python_dispatch import enable_torch_dispatch_mode, push_torch_dispatch_mode, TorchDispatchMode