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

No actual change, just remove variable contain Tensors from global scope (#143225)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/143225
Approved by: https://github.com/ezyang
This commit is contained in:
albanD
2024-12-16 19:20:42 -05:00
committed by PyTorch MergeBot
parent afa313e669
commit 792f1c47e9
12 changed files with 2157 additions and 2138 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
test/test_expanded_weights.py
View File

@ -25,7 +25,11 @@ from torch.testing._internal.common_device_type import (
)
from torch.testing._internal.common_methods_invocations import op_db, SampleInput
from torch.testing._internal.common_modules import module_db, modules
from torch.testing._internal.common_nn import module_tests, new_module_tests, TestBase
from torch.testing._internal.common_nn import (
get_new_module_tests,
module_tests,
TestBase,
)
from torch.testing._internal.common_utils import (
freeze_rng_state,
make_tensor,
@ -1011,7 +1015,7 @@ def filter_supported_tests(t):
# TODO: Once all of these use ModuleInfo, replace with ModuleInfo tests
# These currently use the legacy nn tests
supported_tests = [
t for t in module_tests + new_module_tests if filter_supported_tests(t)
t for t in module_tests + get_new_module_tests() if filter_supported_tests(t)
]
for test_param in supported_tests:
if "constructor" not in test_param: