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

move SchemaCheckMode to torch/_subclasses (#99743)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/99743
Approved by: https://github.com/albanD
This commit is contained in:
Brian Hirsh
2023-04-27 14:50:59 +00:00
committed by PyTorch MergeBot
parent 884c5c86f1
commit 1f2d00e537
3 changed files with 64 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_schema_check.py
View File

@ -8,7 +8,7 @@ import unittest
from torch.testing._internal.common_utils import run_tests
from torch.fx.operator_schemas import normalize_function
from torch.testing._internal.schema_check_mode import SchemaCheckMode
from torch._subclasses.schema_check_mode import SchemaCheckMode
from torch.utils._python_dispatch import TorchDispatchMode
from torch.testing._internal.common_methods_invocations import op_db
from torch.testing._internal.jit_utils import JitTestCase