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

Revert "More markDynamoStrictTest (#115870)"

Browse Source 
This reverts commit 7f686c8fe127cc7db07134297fa09be20ab87918.

Reverted https://github.com/pytorch/pytorch/pull/115870 on behalf of https://github.com/jeanschmidt due to Breaking internal tests and builds, please check diff ([comment](https://github.com/pytorch/pytorch/pull/115870#issuecomment-1862997125))
This commit is contained in:
PyTorch MergeBot
2023-12-19 15:40:57 +00:00
parent 24af118e55
commit a7bfa04da6
4 changed files with 0 additions and 130 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
test/test_autocast.py
View File

@ -8,7 +8,6 @@ from torch.testing._internal.common_utils import TestCase, run_tests, IS_WINDOWS
from torch.testing._internal.autocast_test_lists import AutocastCPUTestLists
from torch.utils._python_dispatch import TorchDispatchMode
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestAutocastCPU(TestCase):
def setUp(self):
super().setUp()
@ -225,7 +224,6 @@ class WeightDTypeCastCounterMode(TorchDispatchMode):
torch.clear_autocast_cache = self.old_clear_cache
return super().__exit__(exc_type, exc_val, exc_tb)
@torch.testing._internal.common_utils.markDynamoStrictTest
@unittest.skipIf(not torch.cuda.is_available(), "requires cuda")
class TestAutocastGPU(TestCase):
def test_cast_cache_is_global(self):
@ -269,7 +267,6 @@ class TestAutocastGPU(TestCase):
torch._C._set_cached_tensors_enabled(False)
@torch.testing._internal.common_utils.markDynamoStrictTest
class TestTorchAutocast(TestCase):
def test_autocast_fast_dtype(self):
gpu_fast_dtype = torch.get_autocast_gpu_dtype()