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

Remove unnecessary skipIfTorchDynamo from test_jit_fuser_te (#118728)

Browse Source 
And add some expected failures.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/118728
Approved by: https://github.com/bdhirsh
This commit is contained in:
rzou
2024-02-12 11:25:34 -08:00
committed by PyTorch MergeBot
parent 28c30f29be
commit 7eecbf8a30
3 changed files with 53 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
test/test_jit.py
View File

@ -344,7 +344,6 @@ class FooToPickle(torch.nn.Module):
self.bar = torch.jit.ScriptModule()
@skipIfTorchDynamo()
class TestJitProfiler(JitTestCase):
"""
This runs tests that requires setting some global states like torch._C._set_graph_executor_optimize
@ -409,7 +408,6 @@ class TestJitProfiler(JitTestCase):
self.assertTrue(other_fn_events[thread] >= mul_time)
@skipIfTorchDynamo()
class TestJit(JitTestCase):
@unittest.skip("Requires a lot of RAM")
def test_big(self):
@ -2944,7 +2942,6 @@ graph(%Ra, %Rb):
self.assertRegex(graph.__repr__(), source_range_regex)
@skipIfTorchDynamo()
class TestFrontend(JitTestCase):
def test_instancing_error(self):
@ -3001,7 +2998,6 @@ class TestFrontend(JitTestCase):
res_2 = traced_model_2(**{'x': torch.rand([2]), 'z': torch.rand([2])}) # noqa: PIE804
@skipIfTorchDynamo()
class TestScript(JitTestCase):
# Tests that calling torch.jit.script repeated on function is allowed.
@ -16041,12 +16037,10 @@ EXCLUDE_ALIAS = {
}
@skipIfTorchDynamo()
class TestJitGeneratedModule(JitTestCase):
pass
@skipIfTorchDynamo()
class TestJitGeneratedFunctional(JitTestCase):
pass