diff --git a/test/run_test.py b/test/run_test.py index 5db4c19288830..72622ab4d7c6a 100755 --- a/test/run_test.py +++ b/test/run_test.py @@ -131,11 +131,10 @@ 'test_autograd', 'test_cpp_extensions_jit', 'test_jit_legacy', + 'test_jit_simple', 'test_dataloader', 'test_overrides', 'test_jit', - 'test_jit_profiling', - 'test_jit_fuser_profiling', 'test_torch', 'distributed/test_distributed', 'distributed/rpc/test_rpc_spawn', diff --git a/test/test_determination.py b/test/test_determination.py index c7ebbb287e6c6..3c8f74a696caa 100644 --- a/test/test_determination.py +++ b/test/test_determination.py @@ -13,7 +13,7 @@ class DeterminationTest(unittest.TestCase): # Test determination on a subset of tests TESTS = [ "test_nn", - "test_jit_profiling", + "test_jit_simple", "test_jit", "test_torch", "distributed/test_distributed", @@ -62,11 +62,11 @@ def test_cpp_file(self): def test_test_file(self): """Test files trigger themselves and dependent tests""" self.assertEqual( - self.determined_tests(["test/test_jit.py"]), ["test_jit_profiling", "test_jit"] + self.determined_tests(["test/test_jit.py"]), ["test_jit_simple", "test_jit"] ) self.assertEqual( self.determined_tests(["test/jit/test_custom_operators.py"]), - ["test_jit_profiling", "test_jit"], + ["test_jit_simple", "test_jit"], ) self.assertEqual( self.determined_tests(["test/distributed/rpc/test_rpc_spawn.py"]),