Add super().setUp() to some test cases (#147651)

I saw that their disabled issues were getting spammed with comments, meaning that they were still running in CI despite having a disable issue, so I added the super().setUp() call to check if there's a disable issue for them since they were missing it

Pull Request resolved: https://github.com/pytorch/pytorch/pull/147651
Approved by: https://github.com/huydhn
This commit is contained in:
Catherine Lee 2025-02-23 18:21:16 +00:00 committed by PyTorch MergeBot
parent f03e7f3801
commit f47573f70d
2 changed files with 2 additions and 0 deletions

View File

@ -693,6 +693,7 @@ class PyWorkTest(TestCase):
class CompileTest(TestCase):
def setUp(self):
super().setUp()
# Allow testing aoti after torch.compile
torch._inductor.config.triton.store_cubin = True
torch._inductor.config.debug = True

View File

@ -17,6 +17,7 @@ from torch.utils._pytree import tree_leaves
class TestDraftExport(TestCase):
def setUp(self):
super().setUp()
init_torchbind_implementations()
@torch._library.register_fake_class("_TorchScriptTesting::_TensorQueue")