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

3
test/test_cuda.py
View File

@ -64,6 +64,7 @@ if TEST_CUDA:
_cycles_per_ms = None
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestCuda(TestCase):
_do_cuda_memory_leak_check = True
_do_cuda_non_default_stream = True
@ -3401,6 +3402,7 @@ exit(2)
self.assertEqual(rc, "0")
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestCudaMallocAsync(TestCase):
@unittest.skipIf(TEST_CUDAMALLOCASYNC, "setContextRecorder not supported by CUDAMallocAsync")
def test_memory_snapshot(self):
@ -3897,6 +3899,7 @@ def reconstruct_from_tensor_metadata(metadata):
@unittest.skipIf(TEST_CUDAMALLOCASYNC or TEST_WITH_ROCM, "NYI")
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestBlockStateAbsorption(TestCase):
def checkCheckpointedBlock(self, before_block, after_block):

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):

1
test/test_cuda_primary_ctx.py
View File

@ -13,6 +13,7 @@ if not TEST_CUDA:
TestCase = NoTest # noqa: F811
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestCudaPrimaryCtx(TestCase):
CTX_ALREADY_CREATED_ERR_MSG = (
"Tests defined in test_cuda_primary_ctx.py must be run in a process "

1
test/test_cuda_trace.py
View File

@ -16,6 +16,7 @@ if not TEST_CUDA:
TestCase = NoTest # noqa: F811
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestCudaTrace(TestCase):
def setUp(self):
torch._C._activate_cuda_trace()