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

markDynamoStrictTest some more tests (#115857)

Browse Source 
Pull Request resolved: https://github.com/pytorch/pytorch/pull/115857
Approved by: https://github.com/voznesenskym
ghstack dependencies: #115845, #115855, #115856
This commit is contained in:
rzou
2023-12-14 11:00:06 -08:00
committed by PyTorch MergeBot
parent 3477a2ee03
commit 8ddca5aeae
4 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/test_cuda_nvml_based_avail.py
View File

@ -25,6 +25,7 @@ with patch.dict(os.environ, {"PYTORCH_NVML_BASED_CUDA_CHECK": "1"}):
TestCase = NoTest # type: ignore[misc, assignment] # noqa: F811
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestExtendedCUDAIsAvail(TestCase):
SUBPROCESS_REMINDER_MSG = (
"\n REMINDER: Tests defined in test_cuda_nvml_based_avail.py must be run in a process "
@ -65,6 +66,7 @@ class TestExtendedCUDAIsAvail(TestCase):
assert in_bad_fork
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestVisibleDeviceParses(TestCase):
def test_env_var_parsing(self):